报错:NVIDIA A30 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.
搜了一下相关资料,大致意思是:用的显卡的算力是8.0,但是当前pytorch依赖的CUDA版本支持的算力是3.7、5.0、6.0、7.0、7.5
所以需要安装争取的cuda版本
我安装了下面这个,解决了问题:
# CUDA 11.3
conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge
更详细的信息可以参考这个博主的