Update __main__.py

theosanderson · Jul 6, 2022 · 8ba94b6 · 8ba94b6
1 parent 071e6ae
commit 8ba94b6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/chronumental/__main__.py b/src/chronumental/__main__.py
@@ -1,7 +1,7 @@
 
 import os
 import sys
-GPU_REQUESTED = "use_gpu" in sys.argv
+GPU_REQUESTED = "--use_gpu" in sys.argv
 if not GPU_REQUESTED:
     os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
 import datetime