[已解决]pytorch cuda版本安装，救急！！！求解答！！！

我又可以了

我安装的python版本是3.11，cuda版本是12.2.或许是两个版本都是最新的，官网上找不到对应的pytorch版本，这个应该怎么解决。。。

最佳答案

月排行榜 / 总排行榜

C#天下第一

2023-10-14 11:37:27

12.1的就可以，没有问题

跳转到最佳答案楼层

isdkz

您好，您可以尝试安装与您的CUDA版本兼容的PyTorch版本。根据PyTorch官网的文档，PyTorch 1.10.0及以上版本需要CUDA 11.1或更高版本。因此，您可以尝试安装PyTorch 1.9.0版本，该版本需要CUDA 11.0或更高版本。您可以使用以下命令安装：

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

如果您的CUDA版本确实是12.2，您可以尝试安装PyTorch 1.10.0版本，该版本需要CUDA 11.1或更高版本。您可以使用以下命令安装：

pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu112/torch_stable.html

请注意，您需要确保您的GPU支持您安装的CUDA版本。如果您的GPU不支持您安装的CUDA版本，您需要降级CUDA版本或升级GPU。

C#天下第一

12.1的就可以，没有问题