小黑的Python日记:Pytorch安装（Win）

大家好呀，我是小黑喵

今天开始学习使用Pytorch，首先我们安装Pytorch。

———————————————（手动分割）—————————————————

Pytorch GPU安装前准备（仅安装CPU版本跳过此部分）

cdDNN Doc:Documentation

安装CUDA与cuDNN

首先，保证已经装好显卡驱动了。没安装的可以去这里：NV驱动

One of the following supported CUDA versions and NVIDIA graphics driver:

• NVIDIA graphics driver R410 or newer for CUDA 10.0
• NVIDIA graphics driver R390 or newer for CUDA 9.2
• NVIDIA graphics driver R384 or newer for CUDA 9
• NVIDIA graphics driver R377 or newer for CUDA 8

这里小黑是RTX所以选择下载CUDA 10.0:CUDA DOWNLOAD

CUDA DOWNLOAD

选择Win-64-10-local-Download，开始下载。下载完后安装。
下载cuDNN：需要注册，填写问卷后下载对应版本即可。

cuDNN

安装如下：首先解压缩cuDNN到某个文件夹

a.将\cuda\bin\cudnn64_7.dll复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin（装好的CUDA文件夹下）
b.将 \cuda\ include\cudnn.h复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\include
c.将 \cuda\lib\x64\cudnn.lib 复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64.
接下来：
a.打开 运行，即搜索-run
b.键入control sysdm.cpl确定，转移到高级，选择环境变量。
c.确保以下变量被设置:
Variable Name: CUDA_PATH
Variable Value: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0

Variable

至此，准备完成。
———————————————（手动分割）—————————————————

安装Pytorch

登录官网：Pytorch Website
点击Get Started

Pytorch Download

选择适合的配置，打开cmd通过给出的pip3下载即可。

pip3

这里小黑嫌下载太慢就自己下载了，再用pip安装。
检验一下：

test

大功告成啦！
接下来就可以正式学习使用pytorch了。

一起学习吧