不能很好的平衡双显卡调用，所以暂时只指定使用1号显卡

2024-06-19 16:06:28 +08:00
parent 585099521b
commit d2c5899122
1 changed files with 1 additions and 1 deletions
--- a/main.py
+++ b/main.py
@@ -9,7 +9,7 @@ from config.log import LOGGING_CONFIG
 from config.photo_review import RETRY_TIME, SEND_ERROR_EMAIL
 from photo_review.photo_review import main

-os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"
+os.environ["CUDA_VISIBLE_DEVICES"] = "1"

 # 项目必须从此处启动，否则代码中的相对路径可能导致错误的发生
 if __name__ == '__main__':