解决安装torch后,torch.cuda.is_available()结果为false的问题
更新时间：2022年12月14日 10:57:52 作者：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
上官网查看对应的安装命令，然后替换掉现有的版本即可，如下
找到自己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的问题
- 第一种方法是一种万金油方法比较推荐，但是安装速度比较感人
- 第二种方法是精准修复，但是需要自己查好对应的版本
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
PyQt5 是一个用于创建图形用户界面的 Python 绑定库,它基于 Qt5 应用程序框架,在 PyQt5 中,QPushButton 是一个常用的控件,用于创建按钮,允许用户通过点击来触发某些操作,本文将详细介绍 QPushButton 的用法,并通过实际案例来展示其强大的功能2024-07-07
这篇文章主要介绍了Python实现的微信公众号群发图片与文本消息功能,结合实例形式详细分析了Python调用微信接口实现微信公众号群发图片与文本消息的具体操作步骤与相关注意事项,需要的朋友可以参考下2017-06-06
这篇文章主要介绍了Python数据分析之pandas函数详解,文中有非常详细的代码示例,对正在学习python的pandas函数的小伙伴们有很好地帮助,需要的朋友可以参考下2021-04-04
python软件测试Jmeter性能测试JDBC Request(结合数据库)的使用详解
这篇文章主要介绍了python软件测试Jmeter性能测试JDBC Request(结合数据库)的使用详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2021-01-01
这篇文章主要介绍了python中自带的三个装饰器的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-11-11
这篇文章主要为大家介绍了python自动化测试selenium执行js脚本的实现示例，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步2021-11-11
这篇文章主要介绍了Python中处理时间的几种方法,包括时间的获取和时间之间的转换等等,需要的朋友可以参考下2015-04-04
这篇文章主要给大家介绍了Python对区间正整数内所有素数之和的相关资料，文中介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-10-10
本文主要介绍了Numpy随机抽样的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-03-03
pandas把dataframe转成Series,改变列中值的类型方法
下面小编就为大家分享一篇pandas把dataframe转成Series,改变列中值的类型方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-04-04
最新评论