nvidia-smi可以显示显卡信息,但torch.cuda.is_available()一直显示false
根本原因是cudatoolkit版本和pytorch版本和cuda版本不对应?1. 查看cuda版本cat /usr/local/cuda/version.txt参考:Linux 和 Windows 查看 CUDA 和 cuDNN 版本2. 找对应的cudatoolkit版本参考binary-compatibility找到对应的 cudatoolkit对应的版本进行安装 。3. 安装对应的cudatoolkit这里的10.0要换成上文中查找到的对应的cuda版本。conda insta
原创
2020-09-08 10:59:35 ·
3388 阅读 ·
0 评论