本文内容: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 | E:\Anaconda (Python需要) |
注:切记记得点击 确定 !!!
五、安装测试:
win+R,输入cmd,进入命令提示符
输入
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 | C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin |
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 | conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ |
三、添加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 | conda uninstall pytorch |