diff --git a/environments/windows/src/tflow.py b/environments/windows/src/tflow.py index a661906..2d9b2da 100644 --- a/environments/windows/src/tflow.py +++ b/environments/windows/src/tflow.py @@ -1,4 +1,10 @@ import tensorflow as tf + +# 36 - optimize for OSX cpu (not gpu) parallelism +os.environ["OMP_NUM_THREADS"] = "10" +tf.config.threading.set_inter_op_parallelism_threads(0) +tf.config.threading.set_intra_op_parallelism_threads(0) + #import keras #from keras.utils import multi_gpu_model #import keras.backend as k