音乐网站建设方案it培训学校it培训机构

方法1

基本上这个就能解决了！！！
在train.py的最上方加上下面这两行，注意是最上面，其次指定的就是你要使用的显卡

import os
os.environ['CUDA_VISIBLE_DEVICES']='6'

方法2：

**问题：**命令行参数指定的是4号卡，但实际却总是在0号卡建立进程
真抽象啊，这一步，模型被送到0号卡，但实际上，送到了4号卡（进程是在4号卡上建立的）
解决办法：
在train.py的train()函数下，第一行选择显卡：

device=torch.device('cuda:4')

自己指定显卡即可，也可以在命令行添加一遍，不冲突，但要一样的

基本上方法一就能解决了。