培训网站建设方案淘宝优化

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

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

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