本文内容:1、Windows下深度学习环境搭建,涉及到anaconda、cuda、cudnn、pytorch安装。

维护:吴海鹏(2020秋入学硕士生)

部署:郭伏雨(博士生)

Windows下深度学习环境搭建

配置环境总体思路:依据python版本选择对应Anaconda版本;

依据显卡驱动版本选择对应的CUDA版本;

依据CUDA版本选择对应的cudnn和pytorch版本。

实验室目前配置的是:python3.7版本(有已下好的搭建环境所需软件)

第一步:Anaconda安装

一、由于anaconda自带Python,只需确定要安装的Python版本,然后下载对应的anaconda版本安装包。

二、下载地址

​ 1.官网 https://www.anaconda.com/products/individual

​ 2.清华源 https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

​ 注:若官网下载速度慢,可用清华源下载;注意操作系统的位数(一般是64位的)

三、安装教程:

​ 1.双击运行下载的.exe文件,按照以下顺序点击:Next → I Agree

​ 2.选择用户:Just Me 或 All Users 都可以,默认选择 Just Me

​ 3.选择安装目录:一般不建议安装到C盘;可更改安装路径,注意安装路径不能出现中文、空格,使用纯英文即可。

​ 4.勾选安装参数:一是添加anaconda到环境变量中(path),二是将anaconda的python版本作为默认的python版本; 建议两者都勾选,环境变量手动再添加一遍(后续有介绍)。

​ 5.等待安装即可,结束时最后 “两个learn” 都不需要勾选,点击Finish即可。

四、配置环境变量

此电脑——属性——高级系统设置——环境变量——path——编辑——新建

​ 1.鼠标选择”此电脑“,单击鼠标右键,选择 ”属性“;

​ 2.左键单击左侧一栏的 “高级系统设置”;

​ 3.点击 “高级”一栏中的 “环境变量”;

​ 4.点击 “系统变量”中的 “Path”项,选择编辑;

​ 5.点击 “新建”,然后添加以下类似路径(保证最后路径的文件名称相同就行)

1
2
3
4
5
E:\Anaconda     (Python需要)
E:\Anaconda\Scripts (conda自带脚本)
E:\Anaconda\Library\mingw-w64\bin (使用C with python的时候)
E:\Anaconda\Library\bin (jupyter notebook动态库)
E:\Anaconda\Library\usr\bin (我自己没找到此文件夹,若存在可将此条也加上)

注:切记记得点击 确定 !!!

五、安装测试:

  1. win+R,输入cmd,进入命令提示符

  2. 输入

    1
    conda --version

    若跳出conda版本,则表示anaconda安装成功,且已经配置好环境变量。

第二步:CUDA安装

一、确定版本

控制面板——硬件和声音——NVIDIA控制面板——帮助——系统信息——组件

​ 1.第三行,NVCUDA64.DLL 可查看显卡驱动版本支持的最高CUDA版本;

​ 2.网页搜索相应CUDA版本安装,建议不需要安装太新版本。网页直接可查找;

二、安装步骤

​ 1.双击运行.exe文件后,前两项是默认安装;

​ 2.当选择安装选项时,选择自定义安装,驱动程序组件第一次安装尽量全选;

​ 3.记得留意CUDA安装位置,之后cudnn的安装会用到!!

三、测试

​ 1.win+R,输入cmd,进入命令提示符

​ 2.输入

1
nvcc -V

四、添加至系统变量

​ 1. 添加方法与anaconda的相同:此电脑——属性——高级系统设置——环境变量——path——编辑——新建

​ 2.添加以下类似路径:

1
2
3
4
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\libnvvp

​ 3.有些路径在安装时可能已经添加,手动补充没有的即可

第三步:cuDNN配置

​ 1.进入官网依照cuda版本选择cudnn文件下载(需要注册账号)

https://developer.nvidia.cn/zh-cn/cuda-toolkit

​ 2.解压下载的cudnn文件,会获得三个文件夹:bin、include、lib;

​ 3.将这三个文件夹复制到之前CUDA的安装路径里,最后一个路径名应是cuda版本号

​ 4.CUDA安装时若是默认路径,则一般为:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0

第四步:pytorch安装(使用清华镜像)

一、打开 Anaconda Prompt

二、添加清华镜像

​ 依次输入下列三行代码:

1
2
3
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

三、添加pytorch镜像

​ 接着输入下列代码:

1
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

四、安装pytorch

​ 进入官网 https://pytorch.org/

​ ①在官网根据自己之前安装的环境依次选择,然后复制官网给出的命令语:

Install——stable——Windows——conda——python——10.2 (以cuda10.2为例)

​ ②若按照上列方式官网提供的cuda版本过高,点击上方的 “ install previous versions of pytroch”,选择旧版本

​ ③根据不同的版本搜索自己的下载语句,记住去掉 -c pytorch

1
conda insatll pytorch torchvision torchaudio cudatookit=10.2  (以cuda10.2为例)

五、注意事项

如果以前安装过pytorch或者安装失败过,在安装前执行下面两条语句:

1
2
conda uninstall pytorch
conda uninstall libtorch