CUDA 12.1配置pytorch

Question

电脑支持cuda最高12.2
安装cuda版本11.8，python版本3.10.10
安装pytorch：pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118

一直报这个错，试过安装其他版本也不行
感谢帮忙解答一下

Answer 1

下午好🌅🌅🌅
本答案参考ChatGPT-3.5

问题描述：在电脑支持最高为Cuda12.2的情况下，尝试安装Cuda版本为11.8和Python版本为3.10.10的Pytorch，并通过 pip install 安装torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0，执行时出现以下错误：

AttributeError: module 'torch' has no attribute 'version'

解决方案：

1. 错误原因是torch版本不兼容，所以需要安装兼容的版本。根据Pytorch官网的要求，支持Cuda12.2的Pytorch版本应该为1.9.0以上，因此可以考虑安装1.9.0版本来兼容。
2. 可以尝试使用以下命令进行安装：

pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/cu111/torch_stable.html

其中，cu111表示支持的Cuda版本，需要根据实际情况进行修改。 3. 安装完成后，可以通过以下代码测试是否安装成功：

import torch
print(torch.__version__)

如果没报错，并打印出了正确的版本号，则说明安装成功。

注意事项：

1. 由于本人电脑暂无法测试，安装的torch版本有可能会稍有差异，请根据实际情况进行选择和安装。
2. 如果还是无法解决问题，可以考虑升级电脑的显卡或更换电脑。