一般命令都可以正常使用,但是使用cuda的命令torch.cuda.is_available()
则输出False。
该问题的根本原因是CUDA环境与Torch版本不匹配,因此最直接的解决方式就是使用官方推荐的版本进行适配。
查看本机安装的cuda版本,通过查阅有三个命令
cat /usr/local/cuda/version.txt nvidia-smi nvcc -V
有时候会不幸输出三个不同的版本,这里在虚拟环境下一定以nvcc -V
查到的版本为主,作为CUDA版本信息。例如我的版本是10.1
上官网查看对应的安装命令,然后替换掉现有的版本即可,如下
找到自己CUDA对应的10.1版本
在虚拟环境中进行安装
Done,开始测试
有些人可能只是cudatoolkit版本对不上,如果运气好,只针对cudatoolkit进行版本匹配即可完成,而且安装速度还很快安装命令,记得添加清华源,否则龟速。。。
cudatoolkit=9.0表示你需要对应上的版本
conda install cudatoolkit=9.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
提供了两种解决torch.cuda.is_available()结果为false的问题
以上为个人经验,希望能给大家一个参考,也希望大家多多支持。