9/2/19

DigitalGrainSize · Sep 2, 2019 · 714fae9 · 714fae9
1 parent c0864c8
commit 714fae9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sedinet_utils.py b/sedinet_utils.py
@@ -13,7 +13,7 @@
 
 import gc, os, shutil
 ## use the first available GPU
-os.environ['CUDA_VISIBLE_DEVICES'] = '1'
+os.environ['CUDA_VISIBLE_DEVICES'] = '0'
 
 ## to use the CPU (not recommended):
 #os.environ['CUDA_VISIBLE_DEVICES'] = '-1'