【故障诊断】【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以上。