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 ·
3478 阅读 ·
0 评论