AttributeError: module ‘torch._C‘ has no attribute ‘_cuda_setDevice‘ pip解决方案

【BUG】AttributeError: module 'torch._C' has no attribute '_cuda_setDevice' pip解决方案。

g302915901

1311人浏览 · 2024-03-16 22:59:29

g302915901 · 2024-03-16 22:59:29 发布

问题：AttributeError: module ‘torch._C’ has no attribute ‘_cuda_setDevice’
原因： 安装的torch的版本是CPU版本
解决方案： 在torch版本后面+cuda版本2.0.1+cu117或者2.0.1+cu118

pip install torch==2.0.1+cu117 torchvision==0.15.2+cu117 torchaudio==2.0.2 -f https://download.pytorch.org/whl/torch_stable.html

可以选择的版本：
1.11.0, 1.11.0+cpu, 1.11.0+cu113, 1.11.0+cu115,
1.12.0, 1.12.0+cpu, 1.12.0+cu113, 1.12.0+cu116,
1.12.1, 1.12.1+cpu, 1.12.1+cu113, 1.12.1+cu116,
1.13.0, 1.13.0+cpu, 1.13.0+cu116, 1.13.0+cu117,
1.13.1, 1.13.1+cpu, 1.13.1+cu116, 1.13.1+cu117,
2.0.0, 2.0.0+cpu, 2.0.0+cu117, 2.0.0+cu118,
2.0.1, 2.0.1+cpu, 2.0.1+cu117, 2.0.1+cu118,
2.1.0, 2.1.0+cpu, 2.1.0+cu118, 2.1.0+cu121,
2.1.1, 2.1.1+cpu, 2.1.1+cu118, 2.1.1+cu121,
2.1.2, 2.1.2+cpu, 2.1.2+cu118, 2.1.2+cu121,
2.2.0, 2.2.0+cpu, 2.2.0+cu118, 2.2.0+cu121,
2.2.1, 2.2.1+cpu, 2.2.1+cu118, 2.2.1+cu121

Triton中文社区

欢迎来到由智源人工智能研究院发起的Triton中文社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。

更多推荐

torch.device(‘cuda:0‘ if cuda else ‘cpu‘)

# 这个地方的cuda:0 实际上并不是0号GPU，他取决于CUDA_VISIBLE_DEVICES# 然后逻辑GPU和物理GPU有一个对应关系# 如果CUDA_VISIBLE_DEVICES为2,1,3# 那么CUDA:0就是2号GPU， CUDA:1 就是1号GPU CUDA:3 就是3号GPUreturn torch.device('cuda:0' if cuda else 'cpu')..