python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > 安装torch后,torch.cuda.is_available()结果为false

解决安装torch后,torch.cuda.is_available()结果为false的问题

作者:l8947943

这篇文章主要介绍了解决安装torch后,torch.cuda.is_available()结果为false的问题,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教

1. 在conda虚拟环境中安装了torch

一般命令都可以正常使用,但是使用cuda的命令torch.cuda.is_available()则输出False。

2. 经过一番查阅资料后

该问题的根本原因是CUDA环境与Torch版本不匹配,因此最直接的解决方式就是使用官方推荐的版本进行适配。

3. 解决思路

查看本机安装的cuda版本,通过查阅有三个命令

cat /usr/local/cuda/version.txt

nvidia-smi

nvcc -V

有时候会不幸输出三个不同的版本,这里在虚拟环境下一定以nvcc -V查到的版本为主,作为CUDA版本信息。例如我的版本是10.1

上官网查看对应的安装命令,然后替换掉现有的版本即可,如下

https://pytorch.org/

找到自己CUDA对应的10.1版本

在虚拟环境中进行安装

Done,开始测试

4. 其他解决思路

有些人可能只是cudatoolkit版本对不上,如果运气好,只针对cudatoolkit进行版本匹配即可完成,而且安装速度还很快安装命令,记得添加清华源,否则龟速。。。

cudatoolkit=9.0表示你需要对应上的版本

conda install cudatoolkit=9.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/

5. 总结

提供了两种解决torch.cuda.is_available()结果为false的问题

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

您可能感兴趣的文章:
阅读全文