【BUG】PyTorch AttributeError: module ‘torch._C‘ has no attribute ‘_cuda_setDevice‘
如果时Python文件中调用了GPU，那么设置：torch.cuda.set_device(-1)，注意-1才代表不使用GPU。
·
原因是：下载的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
欢迎来到FlagOS开发社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
35
0- 0
已为社区贡献1条内容
所有评论(0)