https://stackoverflow.com/questions/52895211/how-do-dask-threads-inte…

…ract-with-openblas-mkl
aertslab · Nov 26, 2018 · 8c3ab67 · 8c3ab67
1 parent b8a3176
commit 8c3ab67
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/pyscenic/cli/pyscenic.py b/src/pyscenic/cli/pyscenic.py
@@ -1,7 +1,14 @@
 # -*- coding: utf-8 -*-
 
-import argparse
 import os
+
+# Set number of threads to use for OpenBLAS.
+os.environ["OPENBLAS_NUM_THREADS"] = 1
+
+# Set number of threads to use for MKL.
+os.environ["MKL_NUM_THREADS"] = 1
+
+import argparse
 import logging
 from dask.diagnostics import ProgressBar
 from multiprocessing import cpu_count