#36 - Update tflow.py for ARM64 parallelism

ObrienlabsDev · Dec 3, 2024 · 3f64906 · 3f64906
1 parent 024117f
commit 3f64906
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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