From 663ed0e704ecde72d9fc5ebd9f72f7450870924e Mon Sep 17 00:00:00 2001
From: Han Zhu <zhuhan568@outlook.com>
Date: Tue, 21 May 2024 05:42:53 -0400
Subject: [PATCH] change device order, fix bug in argument parsing

---
 CryoREAD/main.py | 1 +
 main.py          | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/CryoREAD/main.py b/CryoREAD/main.py
index dc1ca79..d4b3374 100644
--- 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
index 2a6c4c0..b68e944 100644
--- 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"