我在pytorch官网上只找到11.8版本,但是我的cuda是12.0
时间: 2023-05-31 13:01:49 浏览: 168
如果您的CUDA版本是12.0,您需要使用PyTorch的CUDA 11.1版本,因为PyTorch目前不支持CUDA 12.0。
您可以在以下链接中找到PyTorch CUDA 11.1版本的安装指南和下载链接:https://pytorch.org/get-started/locally/
相关问题
我环境中pytorch版本是11.8,但是我的cuda版本是10.1,不兼容,我下载了12.0的cuda怎么在我的环境中用
你可以考虑升级你的pytorch版本,使其与cuda 12兼容。也可以尝试手动编译pytorch以支持你的cuda版本。例如,使用conda或pip安装pytorch时,添加正确的cuda版本和cudatoolkit版本。更多关于pytorch和cuda版本兼容性的信息,可以在pytorch官方文档中找到。
cuda12.0能用pytorch11.8版本
不可以。PyTorch的版本必须与CUDA版本兼容才能正常运行。具体来说,PyTorch 1.8需要CUDA 11.1或更高版本,PyTorch 1.7需要CUDA 11.0或更高版本,PyTorch 1.6需要CUDA 10.2或更高版本。因此,如果你想使用PyTorch 1.8,需要安装CUDA 11.1或更高版本。
阅读全文