pytorch 中注意cuda版本和gpu算力匹配

最新推荐文章于 2025-04-02 19:13:06 发布

m0_46483236

最新推荐文章于 2025-04-02 19:13:06 发布

阅读量2.6w

点赞数 16

分类专栏： pytorch 文章标签： pytorch

本文链接：https://blog.csdn.net/m0_46483236/article/details/124112298

版权

pytorch 专栏收录该内容

28 篇文章

订阅专栏

本文介绍了在安装PyTorch时遇到的CUDA版本兼容性问题。当使用NVIDIA A100 GPU时，由于其CUDA算力为8.0，而当前PyTorch版本仅支持到7.5，导致程序无法正常运行。解决方案是升级到支持CUDA 8.0的cuda11及更高版本，并安装相应兼容的PyTorch。通过在PyTorch官网选择合适的版本，如v1.7.1与CUDA11.0，问题得到解决。参考链接提供了详细步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在安装torch时，一定要注意显卡的cuda版本问题。

比如，在 RTX2080上同样的环境中程序可以正常运行，而换到A100中，就会报错如下：

NVIDIA A100-PCIE-40GB with CUDA capability sm_80 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70 sm_75. If you want to use the NVIDIA A100-PCIE-40GB GPU with PyTorch, please check the instructions at Start Locally | PyTorch

大概意思就是： NVIDIA A100-PCIE-40GB 带有的CUDA算力是8.0，它和现有的PyTorch版本不匹配，现有的PyTorch版本支持的CUDA算力是 3.7，5.0，6.0，7.0，7.5。

支持的CUDA算力是与安装的cuda的版本有关的，cuda 10.2 仅仅支持 3.7，5.0，6.0，7.0算力，不支持8.0算力。而cuda11是支持8.0算力的。

目前安装的torch版本是1.7.0，所以，需要安装cuda11及其以上，并且和torch 1.7.0不冲突的版本。

进入 PyTorch官网：Previous PyTorch Versions | PyTorch

选择合适的CUDA版本，也可以去 Previous PyTorch Versions 进行查看选择，

最终选择了 v1.7.1 CUDA 11.0的版本

# CUDA 11.0
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

问题解决。

参考：https://zhuanlan.zhihu.com/p/427395039