conda下安装pytorch最详细教程 // 安装pytorch踩坑记录 // cuda11.5下pytorch安装 // torch.cuda.is_available()输出False解决办法

几点说在前面!!!!!

我踩的坑:

1、一开始入坑使用pytorch框架没有用anaconda,现在非常后悔!!!conda对小白管理环境真的巨好用!!!
2、安装时候torch版本不对应,cuda版本不对应,python版本不对应,等等!!反正就是不好用!!很恼火!
3、pip安装包中断,安装速度慢!!!
4、各种版本问题,调用安装的包里类和方法等报错,不能正常运行!!!
5、等等各种问题。

今天实在是蚌埠住了,删了所有的环境和python重新安装,然后吸取教训一次成功,下面分享我的方法:

一、首先安装anaconda:

各位可以移步我的这一篇博客进行安装:
https://blog.csdn.net/qq_45281807/article/details/112442577
也可以去官网安装哦!

二、配置cuda驱动软件

1、首先查看自己电脑的NVIDIA显卡型号

同时按下Ctrl+Shift+Tab键进入任务管理器,如图所示查看GPU型号:

conda下安装pytorch最详细教程 // 安装pytorch踩坑记录 // cuda11.5下pytorch安装 // torch.cuda.is_available()输出False解决办法
https://www.nvidia.com/Download/index.aspx?lang=cn
小编为笔记本NVIDIA GeForce GTX 1050 GPU。
conda下安装pytorch最详细教程 // 安装pytorch踩坑记录 // cuda11.5下pytorch安装 // torch.cuda.is_available()输出False解决办法
点击搜索,下载你自己显卡的最新的驱动。
之后进行安装。

; 2、安装CUDA驱动

进入官网下载驱动:
https://www.geforce.com/hardware/technology/cuda/supported-gpus
点击立即下载:

conda下安装pytorch最详细教程 // 安装pytorch踩坑记录 // cuda11.5下pytorch安装 // torch.cuda.is_available()输出False解决办法
conda下安装pytorch最详细教程 // 安装pytorch踩坑记录 // cuda11.5下pytorch安装 // torch.cuda.is_available()输出False解决办法

; 3、查看驱动软件是否安装成功

win+R输入cmd进入控制窗:
输入:

nvcc -V

查看已安装的版本,如运行出以下结果则安装成功!(我安装的是cuda11.5)

conda下安装pytorch最详细教程 // 安装pytorch踩坑记录 // cuda11.5下pytorch安装 // torch.cuda.is_available()输出False解决办法

三、安装pytorch

1、首先创建环境

打开anaconda软件,按下图所示点击进入:

conda下安装pytorch最详细教程 // 安装pytorch踩坑记录 // cuda11.5下pytorch安装 // torch.cuda.is_available()输出False解决办法
conda下安装pytorch最详细教程 // 安装pytorch踩坑记录 // cuda11.5下pytorch安装 // torch.cuda.is_available()输出False解决办法
点击create创建完成。

; 2、添加加速镜像源通道

从官网安装真的非常非常慢!!!
并且会出现中断、包不全等各种你想不到的报错!!!
从开始选择这个点击打开:

conda下安装pytorch最详细教程 // 安装pytorch踩坑记录 // cuda11.5下pytorch安装 // torch.cuda.is_available()输出False解决办法
conda activate pytorch2(这里换成你的环境名字哈)

conda下安装pytorch最详细教程 // 安装pytorch踩坑记录 // cuda11.5下pytorch安装 // torch.cuda.is_available()输出False解决办法
依次输入下列代码添加镜像源,这里给大家推荐5个好用的源,建议全部添加!!:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/cloud/bioconda/

3、输入代码安装pytorch

继续输入:

conda install pytorch torchvision torchaudio cudatoolkit=11.5

然后等一下会显示是否继续进行, 输入y,点回车,就可以安装啦!!
可以看到只要网速没限制,添加过源后的下载安装包的速度还是飞快飞快滴! !

四、测试cuda是否可用

1、 按照如图所示,输入指令,若返回True,则证明cuda可用!!

conda下安装pytorch最详细教程 // 安装pytorch踩坑记录 // cuda11.5下pytorch安装 // torch.cuda.is_available()输出False解决办法

; 3、返回False解决办法

如果你按照以上步骤安装一般是没有问题的。
如果出现False,大家可以尝试以下步骤进行排除:
1、首先确定显卡是否支持CUDA,https://www.geforce.com/hardware/technology/cuda/supported-gpus
2、确保驱动升级到最新(确保驱动与cuda版本互相支持)

小结

到这里就全部结束啦!
相信聪明的你一定也安装好了环境,如果你从我的分享中获得了或多或少的帮助,可以点一个赞哦!
小生不才,如有错误,欢迎评论指正!!
欢迎交流!!
私信看到就会回!!

Original: https://blog.csdn.net/qq_45281807/article/details/121294644
Author: 伏城无嗔
Title: conda下安装pytorch最详细教程 // 安装pytorch踩坑记录 // cuda11.5下pytorch安装 // torch.cuda.is_available()输出False解决办法

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/710542/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球