A100-PCIE-40GB with CUDA capability sm_80 is not compatible with the current PyTorch installation

今天跑代码的时候，遇到个提示：

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

这主要是因为之前pytorch是通过conda安装的，A100自带了sm_80，通过conda安装会破坏原始的驱动，因此，解决的方法也很简单，只需要利用pytorch的本地安装方法进行安装即可~

官方的安装方法如下：pip3 install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

不过，由于国内访问pytorch的官网速度实在太慢，我选择了离线安装torch，然后在线安装torchvision和torchaudio。torch的离线安装包我放在了云盘，需要的文末自取。

下载torch之后，执行如下两条命令即可完成安装：

pip install torch-1.8.1+cu111-cp37-cp37m-linux_x86_64.whl
pip install torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html