原因是：下载的PyTorch是CPU版本的，而不是GPU版本的。
你可以用这两行代码
import torch
torch.cuda.is_available()
检测自己的的cuda是否可用，也就是GPU版本是否存在，（严谨的说应该是检测GPU版本是否安装成功）。
要么下载GPU版本，要么在你执行代码时加上将GPU设置成-1。
- 如果时Python文件中调用了GPU，那么设置：
torch.cuda.set_device(-1)
- 如果你用命令行执行python文件，那么在最后加上
python train.py --你的GPU的设置对应形参 -1
你的GPU设置对应形参，是你的Python文件中应该会有一个arg是用来设置要使用GPU的编号的，与1同理。
参考感谢：
[1] https://blog.csdn.net/weixin_39450145/article/details/104797786