多GPU训练

要在具体用加载模型代码的文件里面加上
import os

os.environ["CUDA_VISIBLE_DEVICES"] = "1"
才有效果

 

posted on 2021-09-09 15:23  cltt  阅读(37)  评论(0)    收藏  举报

导航