Echo/fcb_photo_review
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
liuyebo
2024-06-19 16:06:28 +08:00
parent 585099521b
commit d2c5899122
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
main.py
View File

@@ -9,7 +9,7 @@ from config.log import LOGGING_CONFIG
from config.photo_review import RETRY_TIME, SEND_ERROR_EMAIL from config.photo_review import RETRY_TIME, SEND_ERROR_EMAIL
from photo_review.photo_review import main from photo_review.photo_review import main
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1" os.environ["CUDA_VISIBLE_DEVICES"] = "1"
# 项目必须从此处启动，否则代码中的相对路径可能导致错误的发生 # 项目必须从此处启动，否则代码中的相对路径可能导致错误的发生
if __name__ == '__main__': if __name__ == '__main__':