change device order, fix bug in argument parsing

kiharalab · May 21, 2024 · 663ed0e · 663ed0e
1 parent 930e373
commit 663ed0e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/CryoREAD/main.py b/CryoREAD/main.py
@@ -25,6 +25,7 @@ def init_save_path(origin_map_path):
 
         gpu_id = params["gpu"]
         if gpu_id is not None:
+            os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
             os.environ["CUDA_VISIBLE_DEVICES"] = gpu_id
         cur_map_path = os.path.abspath(params["F"])
         # process the map path if it's ending with .gz

diff --git a/main.py b/main.py
@@ -17,7 +17,7 @@
 
     args = parser.parse_args()
 
-    logger.info("Input job folder path: " + args.F)
+    logger.info("Input job folder path: ", args.F)
 
     CRYOREAD_PATH = "./CryoREAD/main.py"