今天开始学习使用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
选择Win-64-10-local-Download,开始下载。下载完后安装。
下载cuDNN:需要注册,填写问卷后下载对应版本即可。
安装如下:首先解压缩cuDNN到某个文件夹<installpath>
a.将<installpath>\cuda\bin\cudnn64_7.dll复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin(装好的CUDA文件夹下)
b.将 <installpath>\cuda\ include\cudnn.h复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\include
c.将 <installpath>\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
至此,准备完成。
———————————————(手动分割)—————————————————
安装Pytorch
登录官网:Pytorch Website
点击Get Started
选择适合的配置,打开cmd通过给出的pip3下载即可。
这里小黑嫌下载太慢就自己下载了,再用pip安装。
检验一下:
大功告成啦!
接下来就可以正式学习使用pytorch了。