【BUG】PyTorch AttributeError: module ‘torch._C‘ has no attribute ‘_cuda_setDevice‘

原因是：下载的PyTorch是CPU版本的，而不是GPU版本的。

你可以用这两行代码

import torch
torch.cuda.is_available()

检测自己的的cuda是否可用，也就是GPU版本是否存在，（严谨的说应该是检测GPU版本是否安装成功）。

要么下载GPU版本，要么在你执行代码时加上将GPU设置成-1。

1. 如果时Python文件中调用了GPU，那么设置：

torch.cuda.set_device(-1)

2. 如果你用命令行执行python文件，那么在最后加上

python train.py --你的GPU的设置对应形参 -1

你的GPU设置对应形参，是你的Python文件中应该会有一个arg是用来设置要使用GPU的编号的，与1同理。

参考感谢：

[1] https://blog.csdn.net/weixin_39450145/article/details/104797786