minor edit

amaiya · Mar 18, 2020 · 25d40f1 · 25d40f1
1 parent 9a2b98a
commit 25d40f1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ktrain/tests/testenv.py b/ktrain/tests/testenv.py
@@ -3,6 +3,6 @@
 #os.environ['TF_EAGER'] = '0'
 import sys
 os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID";
-os.environ["CUDA_VISIBLE_DEVICES"]="1"
+os.environ["CUDA_VISIBLE_DEVICES"]="0"
 sys.path.insert(0,'../..')