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"