conda下安装pytorch最详细教程安装pytorch踩坑记录cuda11.5下pytorch安装torch.cuda.is

conda下安装pytorch最详细教程安装pytorch踩坑记录cuda11.5下pytorch安装torch.cuda.is,第1张

conda下安装pytorch最详细教程 // 安装pytorch踩坑记录 // cuda11.5下pytorch安装 // torch.cuda.is 几点说在前面!!!!! 我踩的坑:

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型号:
然后进入NVIDIA官网选择自己GPU型号的最新驱动,进行下载安装:
https://www.nvidia.com/Download/index.aspx?lang=cn
小编为笔记本NVIDIA GeForce GTX 1050 GPU。

点击搜索,下载你自己显卡的最新的驱动。
之后进行安装。

2、安装CUDA驱动

进入官网下载驱动:
https://www.geforce.com/hardware/technology/cuda/supported-gpus
点击立即下载:
根据提示根据自己的系统选择,点击右下角Download,然后根据提示安装。

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

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

nvcc -V

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

三、安装pytorch 1、首先创建环境

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

自己起环境名字,目前建议选择python3.7,兼容性好:

点击create创建完成。

2、添加加速镜像源通道

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

conda activate pytorch2(这里换成你的环境名字哈)


依次输入下列代码添加镜像源,这里给大家推荐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可用!!

3、返回False解决办法

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

小结

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

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5479707.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-12
下一篇 2022-12-12

发表评论

登录后才能评论

评论列表(0条)

保存