A100与pytorch版本不兼容问题

前提

在A100服务器上用使用torch1.9.1+cu102

报错信息

NVIDIA A100-SXM4-80GB 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.
If you want to use the NVIDIA A100-SXM4-80GB GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

RuntimeError: CUDA error: no kernel image is available for execution on the device
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.

解决

NVIDIA A100-PCIE-40GB 带有的CUDA算力是8.0，它和现有的PyTorch版本不匹配。因此将CUDA版本提高到11.0以上。
在官网pytorch查询如下命令：

pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1 -f https://download.pytorch.org/whl/cu111/torch_stable.html

重新运行代码，问题解决。
参考文章：
[1]:https://blog.csdn.net/weixin_46065198/article/details/132295334