【故障诊断】【Ubuntu服务器】NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the curre

【故障诊断】【Ubuntu服务器】NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the curreent PyTorch installation.

报错信息

报错信息

故障解读

我们的显卡是3090,因此算力是sm86,也就是说,算力8.6。但是当前的PyTorch依赖的CUDA版本支持的算力只有3.7、5.0、6.0、7.0。支持的最高算力达不到8.6,所以我们只能重装pytorch。那么这时候有人会问了,你的cuda也不对呀,不过这时候我们可以看看cuda的版本是多少,使用命令nvcc --version就可以去看cuda版本号了,经查勘,我们的cuda是11.1,可以支持sm86的算力。因此我们只需要更新pytorch即可。

另:RTX3090的显卡算力十足,相对应的cuda必须要达到11.0以上。

猜你喜欢

转载自blog.csdn.net/Michael_Cretu_/article/details/123191337
今日推荐