0


Win11系统PyTorch环境安装配置

1.安装anaconda

anaconda官网下载地址:Anaconda | Anaconda Distribution

下载完成后双击exe应用程序,开始进行安装,详细过程请看另一位博主文章:超详细Anaconda安装教程_极小白的博客-CSDN博客_anaconda

注:安装位置可选其他盘,如果害怕记不清位置也可默认C盘,安装时记得勾选自动添加环境变量项,这样可以减少手动添加的麻烦,其他均按照默认next下一步直至安装完成。anaconda安装时自带的有python,我们后续也可以在anaconda创建一个虚拟环境安装自己需要的python版本,不同环境中的python等包并不冲突,互相独立。

检验是否安装成功:

win+r输入cmd打开命令窗口

输入指令:

conda
![](https://img-blog.csdnimg.cn/f415383b95c54d39b0afc676997c02f8.png)

查看版本号:

conda -V
![](https://img-blog.csdnimg.cn/07031f05645a4609a2df166b15dd47ea.png)

如上所示即为安装成功。

2.安装cuda

下载安装之前,一定要先查看自己电脑所能支持的cuda最高版本是多少,不要超过这个版本!!!

首先打开NVIDIA控制面板,点击左下角系统信息——》组件进行查看

接下来进入cuda官网进行下载:CUDA Toolkit - Free Tools and Training | NVIDIA Developer

进入后点击DOWNLOAD NOW进入如下界面:

注意:这里官方提供的是最新版的cuda,建议不要下载最新,因为后续配置torch环境时会出现版本不匹配等一系列问题。。。

因此,我们可以去历史版本页面进行下载:CUDA Toolkit Archive | NVIDIA Developer

进去之后选择我们所需要的cuda版本进行下载即可,笔者接下来以cuda11.1.0版本为例进行安装介绍

我的电脑是win11系统,点开cuda11.1.0版本会发现没有win11选项只有win10,这里我们直接选择win10即可,放心大胆的下载安装,笔记亲身安装过不会有问题。但是win11不支持10.x版本cuda!!!

双击exe文件进行安装

具体安装过程请看另一位博主介绍,我当时也是看的这篇文章,介绍的很仔细:CUDA安装教程(超详细)_Billie使劲学的博客-CSDN博客_cuda安装

安装结束后,我们需要查看系统环境变量path下是否自动添加,若没有需要我们手动添加,有以下四个:

确认环境变量无误后,验证cuda是否安装成功:

cmd进入命令窗口,输入

nvcc -V  #查看版本号
set cuda  #查看环境变量

如上所示即安装成功。

3.安装cudnn

cudnn是用于配置深度学习使用,官网下载地址:https://developer.nvidia.com/rdp/cudnn-download

注意:第一次需要我们进行注册登录NVIDIA账户

这里我们一定要选择和自己cuda版本相匹配的cudnn进行下载!!!!

这里我选择的是8.4.0版本,因为它支持11.x版本的cuda。

下载完成后,是一个压缩包,解压后打开里面是这样几个文件:

我们需要把以上红框中的三个目录copy到cuda相应安装目录下进行替换,详细请看这位博主的介绍:CUDA安装教程(超详细)_Billie使劲学的博客-CSDN博客_cuda安装

替换完成后,去查看系统环境变量path下是否有以下四个变量,若没有需要手动进行添加

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\include

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\lib

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\libnvvp

接下来验证安装是否完成:

首先win+R启动cmd,cd到安装目录下的 …\extras\demo_suite,然后分别执行bandwidthTest.exe和deviceQuery.exe,如下图所示:

至此,cudnn也安装完成。

4.安装pytorch

pytorch官网下载地址:PyTorch

这里一定一定一定要找到和自己cuda相匹配的torch版本进行安装,我们可以在历史版本中进行查看,这里我选择的是1.9.1版本,对应cuda11.1.0

复制以上下载指令,这里建议不要用conda命令,最好用pip下载,因为conda默认是cpu版本,我们要用的是gpu版本。

找好所需要下载安装的torch版本后,接下来开始进入正戏,进行安装:

在windows搜索栏搜索anaconda找到如下应用,点击打开,会弹出类似cmd的窗口

前面带base的是anaconda中系统的环境,我们最好不要在base中安装配置,可以自己创建一个虚拟环境在里面进行安装配置,各个虚拟环境之间相互独立不受影响,这正是anaconda的方便之处,上图中的pytorch191就是我创建的一个虚拟环境。

接下来,我们首先创建一个虚拟环境取名pytorch001,在创建时我们可以指定环境内python版本,这里选择python3.7(python3.7和大部分版本匹配)

  1. 创建虚拟环境``````conda create -n pytorch001 python=3.7
  2. 创建完成后,查看当前anaconda中存在那些环境conda env list或者conda info -e可以看到pytorch001已经创建成功3.激活虚拟环境activate pytorch001``````![](https://img-blog.csdnimg.cn/c3741d2007e84b938fb51d3c7bbca8d0.png)可以看到激活后我们就进入了pytorch001环境下,此时我们就可以在这个环境下安装一些东西了4.查看当前环境下python版本python -V 5.安装torch,复制之前在官网粘贴的下载指令pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html进行下载安装,这个过程可能需要一些时间,我也可以使用pip镜像源,这样能快一点# 配置中科大镜像pip config set global.index-url https://mirrors.ustc.edu.cn/pypi/web/simple# 配置阿里源pip config set global.index-url https://mirrors.aliyun.com/pypi/simple/# 配置腾讯源pip config set global.index-url http://mirrors.cloud.tencent.com/pypi/simple# 配置豆瓣源pip config set global.index-url https://pypi.douban.com/simple/查看当前镜像源pip config list以上镜像源,选择任意一个均可如上所示,即安装成功。 接下来需要进行验证检查:pytorch安装检验``````import torch print(torch.__version__) print(torch.cuda.is_available()) 如上所示即安装成功,且gpu加速开启成功。 注意:有的人可能会出现安装失败的问题,多数情况下是cuda,cudnn,python和torch版本不匹配的问题。还有一些人是gpu显示false没有开启,那么很大原因是上面我提到的,是使用conda指令下载的cpu版本的torch,所以一定要看清楚,选择gpu版本用pip下载最好。至此,pytorch环境配置全部完成!!!还有一些anaconda中常用的指令,请看我的另一篇文章:Anaconda常用命令及pip镜像源_梦想就是摸鱼的博客-CSDN博客

参考文章:

超详细Anaconda安装教程_极小白的博客-CSDN博客_anaconda

CUDA安装教程(超详细)_Billie使劲学的博客-CSDN博客_cuda安装

pytorch安装教程 - 知乎


本文转载自: https://blog.csdn.net/qq_36791745/article/details/125388713
版权归原作者 梦想就是摸鱼 所有, 如有侵权,请联系我们删除。

“Win11系统PyTorch环境安装配置”的评论:

还没有评论