我又可以了 发表于 2023-9-15 22:22:04

pytorch cuda版本安装，救急！！！求解答！！！

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

isdkz 发表于 2023-9-15 22:22:18

您好，您可以尝试安装与您的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#天下第一 发表于 2023-10-14 11:37:27

12.1的就可以，没有问题
页: [1]
查看完整版本: pytorch cuda版本安装，救急！！！求解答！！！