方法1
基本上这个就能解决了!!!
在train.py的最上方加上下面这两行,注意是最上面,其次指定的就是你要使用的显卡
bash
import os
os.environ['CUDA_VISIBLE_DEVICES']='6'
方法2:
**问题:**命令行参数指定的是4号卡,但实际却总是在0号卡建立进程
真抽象啊,这一步,模型被送到0号卡,但实际上,送到了4号卡(进程是在4号卡上建立的)
解决办法:
在train.py的train()函数下,第一行选择显卡:
bash
device=torch.device('cuda:4')
自己指定显卡即可,也可以在命令行添加一遍,不冲突,但要一样的
基本上方法一就能解决了。