Pythonの並列処理ライブラリであるdaskで計算処理を行うと、デフォルトでは最大コア数分のスレッドを使って計算を行います。スレッド数を変えて計算するには次の方法があります。
num_workersで指定
computeメソッドを呼び出すときにnum_workersというキーワード引数で計算の並列数を指定できます。デフォルトではスレッドを使って並列計算を行うので、これでスレッド数が指定できることになります。
import dask.array as da
z = da.random.normal(0, 1, (1000, 1000))
x = da.sum(z, axis=1)
x.compute(num_workers=2) # 2スレッドで計算
dask.configで指定
computeメソッド呼び出し時に個別にスレッド数を指定するのではなく、まとめて指定したい場合には次のようにします。
from multiprocessing.pool import ThreadPool
import dask
dask.config.set(pool=ThreadPool(20))