解决安装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的问题
- 第一种方法是一种万金油方法比较推荐，但是安装速度比较感人
- 第二种方法是精准修复，但是需要自己查好对应的版本
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
《坦克大战》以二战坦克为题材,既保留了射击类游戏的操作性,也改进了射击类游戏太过于复杂难玩的高门槛特点,集休闲与竞技于一身。经典再度袭来,流畅的画面,疯狂的战斗,让玩家再次进入疯狂坦克的世界。玩家的目标是控制坦克躲避危险,消灭掉所有的敌人即可进入下一关2022-02-02
Python matplotlib画图与中文设置操作实例分析
这篇文章主要介绍了Python matplotlib画图与中文设置操作,结合实例形式分析了Python使用matplotlib进行图形绘制及中文设置相关操作技巧,需要的朋友可以参考下2019-04-04
Python IDLE Subprocess Connection Error的简单解决方法
最近用要Python处理一点事,就打开Python IDLE,结果出现错误,下面这篇文章主要给大家介绍了关于Python IDLE Subprocess Connection Error的简单解决方法,需要的朋友可以参考下2023-01-01
这篇文章主要介绍了python目录与文件名操作例子,需要的朋友可以参考下2016-08-08
这篇文章主要为大家详细介绍了python绘制简单彩虹图的方法，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-11-11
这篇文章主要介绍了python中的多重继承实例讲解,本文着重讲解寻找父类的顺序,分为经典类和新式类,需要的朋友可以参考下2014-09-09
大家好，本篇文章主要讲的是 python Tkinter是什么，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下2022-01-01
这篇文章主要介绍了解决python存数据库速度太慢的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2021-04-04
这篇文章主要给大家介绍了软玉Python中常见的序列化操作的相关资料，文中通过示例代码介绍的非常详细，对大家学习或者使用Python具有一定的参考学习价值，需要的朋友们下面来一起学习学习吧2019-06-06
Python+Selenium实现短视频自动上传与发布的实践
本文主要介绍了Python+Selenium实现短视频自动上传与发布的实践，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2022-04-04
最新评论