当前位置：首页 > 教程 > 编程语言 > python编程 > Windows下实现pytorch环境搭建

Windows下实现pytorch环境搭建
日期：2021-09-08 13:58:02 来源：脚本之家

Anaconda安装

安装步骤：

1、官网下载安装包：https://www.anaconda.com/distribution/

2、运行并选择安装路径，等待安装完成。(要记得勾选 Add Anaconda to the system PATH environment variable，是为了将Anaconda添加到环境变量中)

在这里插入图片描述

3、查看是否安装成功，cmd中输入conda回车，是否出现如下信息，有则说明安装成功。

在这里插入图片描述

CUDA与CuDNN安装

一、先检查自己的电脑所支持的CUDA版本是多少。

步骤：

1、桌面右键点击进去NVIDIA控制面板，找到左下角的系统信息，点击组件，出现如下界面。**

从NVCUDA.DLL 这一行后面的CUDA 10.2说明我的电脑所支持的最高版本是10.2。

2、官网下载相对应的CUDA: (https://developer.nvidia.com/cuda-toolkit-archive)

在这里插入图片描述

我所支持的版本是10.2，所以我下载的是红色箭头所标出的那行。点击以后出现如下页面，选择第一个下载即可。

在这里插入图片描述

下载完成后，在所在的文件夹下运行安装即可。
然后查看CUDA是否安装成功：
cmd中运行到安装的文件目录下：

cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin

然后执行nvcc -V，查看是否出现如下信息，有则说明CUDA安装成功。

在这里插入图片描述

如果输入nvcc -V 出现错误，考虑是否将Anaconds加入环境变量PATH中。

在这里插入图片描述

这是我在实现中出现问题的时，所解决的方法，大家如果还是不行可以自己再找找方法。

3、安装相对应版本的cuDNN。

（1）进入官网：
https://developer.nvidia.com/rdp/cudnn-download
需要注册并登录账号（这个需要一段时间），然后选择相对应的版本下载。

在这里插入图片描述

我的是CUDA10.2，所以我选择的是第二个。等待下载完成后进行解压，得到一个cuda文件夹，进入之后，全选，复制到之前CUDA所安装的文件夹下，有重复的进行替换即可。

在这里插入图片描述

查看cuDNN是否安装成功：

步骤如下：（进入安装的路径）

在这里插入图片描述

出现如下Result = PASS 说明cuDNN安装成功。

在这里插入图片描述

可以再接着执行deviceQuery.exe，如果出现Result = PASS 说明CUDA和cuDNN都已经安装成功了。

Pytorch安装

1、先配置torch环境。

cmd中输入：

conda create –n new python=3.7.4

(这儿python=xxx，大家可以先通过在cmd中直接输入python，就会出现你的python的版本，这儿的new是你所创建的虚拟环境的名称，大家可以自己取)
然后激活环境：
cmd中输入:

activate new

2、pytorch的安装

法一：打开pytorch官网：
https://pytorch.org/get-started/previous-versions/
找到自己所对应版本的pip即可（会慢些）。

法二：进入如下网址：https://download.pytorch.org/whl/torch_stable.html，下载自己所需要的torch和torchvision。
可以按快捷键ctrl+F进行搜索。

在这里插入图片描述

（cu102表示的是cuda10.2，大家根据自己cuda的版本进行搜索，cp37是python的版本，我的是3.7.4，win就是windows系统了，选择最新的版本下载即可）
查找torchvision也是同理。下载完成后，
要先进入刚才创建的torch环境中，即输入 activate new。
前面的(bash)就会变成 (new)。
接下来操作如下：

在这里插入图片描述