在安装torch时，一定要注意显卡的cuda版本问题。
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
Original: https://blog.csdn.net/m0_46483236/article/details/124112298
Author: m0_46483236
Title: pytorch 中注意cuda版本和gpu算力匹配
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/706147/
转载文章受原作者版权保护。转载请注明原作者出处！