代码收藏家技术教程 2022-08-02

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型号：
然后进入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版本互相支持）