RTX 3090的算力是8.6,但是当前的PyTorch依赖的CUDA版本支持的算力只有3.7、5.0、6.0、7.0
算力7.0的显卡可以在支持最高算力7.5的CUDA版本下运行,但是算力7.5的显卡不可以在支持最高算力7.0的CUDA版本下运行。同理算力8.x的显卡不可以在支持最高算力7.x的CUDA版本下运行
解决办法:重新安装cuda11.1和cudann8.2.0
https://blog.youkuaiyun.com/m0_37605642/article/details/98854753
查询显卡算力
- 进入deviceQuery.exe文件目录下
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7\extras\demo_suite
- 执行当前目录下的deviceQuery.exe文件
./deviceQuery.exe文件