Force tensorflow to be run on CPU

Steffenhir · Oct 20, 2023 · e574cd9 · e574cd9
1 parent ada157e
commit e574cd9
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/graxpert/background_extraction.py b/graxpert/background_extraction.py
@@ -14,9 +14,10 @@
 from skimage.transform import resize
 from skimage.filters import  gaussian
 
-import tensorflow as tf
-import os
 import sys
+import os
+import tensorflow as tf
+os.environ['CUDA_VISIBLE_DEVICES'] = '-1' # Run tensorflow on CPU
 
 from graxpert.mp_logging import get_logging_queue, worker_configurer
 from graxpert.parallel_processing import executor