背景:
最近在进行某项算法工程部署任务时用到了比较老的torch1.6版本,在更换版本后发现环境出现了各种冲突,首先时torch与cuda的版本冲突,在更换cuda版本后发现显卡(注意这里不是驱动,而是显卡本身)与cuda产生了冲突
环境:
ubuntu20.04
CUDA10.2
torch1.6
显卡RTX A2000
报错内容:
Laptop GPU with CUDA capability sm_86 is not compatible with current PyTorch..
报错原因:
查看显卡算力和其与cuda版本的对应关系:显卡、显卡驱动、cuda、torch四者之间的关系-CSDN博客
cuda版本与显卡算力不匹配,我是A2000的卡,其算力为8.6,sm_86就是算力为86的意思。为了适配torch1.6我安装了10.2的cuda,但是只有11.0以上的c