深度学习环境配置:win10+CUDA10.1+cuDNNv7.6.4+Anaconda3+pytorch+pyCharm编辑器
深度学习环境配置
借鉴了很多博客,自己做个记录。
前言
系统是win10家庭版,显卡是DVIDIA 20系列的RTX2060,cpu是i7 10700k。小白一个,做一点简单的迁移学习,所以配置不需要辣么高啦~
一、CUDA10.1安装
1.去官网下载
cuda10.1
2.安装
AppData是临时存放的隐藏位置,不需要改
安装方式选择精简,然后一直按照流程点,有add path之类的要加。最后默认安装位置为
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
3.验证
windows +R 输入cmd
输入nvcc -V
表示第一步安装CUDA成功。
二、cuDNN7.6.4下载
去官网(下载链接)下载cuDNN7.6.4,注意需要注册一个账号
如图开始选择,使用邮箱注册一个账号
选择自己对应的版本下载,注意CUDA与cuDNN是有版本对应关系的,可自行查找。
下载后将解压后的三个文件夹复制
直接粘贴到CUDA安装路径下
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
原文件夹下也有这三个文件,可选择替换。
验证cuDNN是否成功:
配置完成后,我们可以验证是否配置成功,主要使用CUDA内置的deviceQuery.exe 和 bandwithTest.exe:
首先win+R启动cmd
cd到安装目录下的C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\demo_suite
然后分别执行bandwidthTest.exe和deviceQuery.exe,应该得到下图:
结果都出现Result=PASS,则表示第二步配置cuDNN成功。
三、Anaconda3安装
1.下载
官网下载地址
选择windows版本,我选择的是anaconda3-5.3.1
2.安装
下载后就按照步骤一步步来,注意一定要点**添加路径!!!**如下图所示:
上图表示第三步安装anaconda3 成功。
也可以使用下面的验证方式:
附一个向详细一点的安装教程:
anaconda3+win10
四、pyCharm编辑器下载安装
pyCharm是python的IDE,比较好上手,可能更适合我们女孩子些吧。
去官网下载pyCharm,选择社区版,社区版免费开放,专业版需要激活。
下载后按照步骤一步步安装,记得也要添加路径
finish后就表示第四步pyCharm安装成功啦。
专业版安装教程
五、pytorch配置
整个下载安装配置中,pytorch框架的搭建是最让我头疼的,也是稍微麻烦一点的。
1.创建新的虚拟环境,命名为pytorch:
使用新的虚拟环境:
以上是为了方便创建了一个虚拟环境备用,这一步可以不做。
2.使用清华源镜像下载pytorch
使用官网源下载太慢了,建议使用清华源。
首先在pytorch(官网:https://pytorch.org/)查找要版本对应的下载代码,下图默认pytorch=1.7.0,torchvision=0.8.1:
然后在创建好的虚拟环境窗口下一步步复制代码,注意最后一行是pytorch官网下载地址,**已经去掉了 -c pytorch!!!**否则还是按照官网源速度下载,特别慢还特别容易出问题。
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
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda install pytorch torchvision cudatoolkit=10.1
如下图所示:
3.下载完成验证如下图:
cmd 输入python
然后输入 import torch
结果不保存,则下载成功
即表示第五步pytorch安装完成,因为我使用的IDE是pyCharm,所以还需要在编辑器中搭建pytorch框架。
六、在pyCharm中使用pytorch框架
选择create project
选择上图步骤,ok应用,搭建pytorch框架成功!
小结:
这是第二次进行环境配置了,感觉还是会出现有很多问题,借鉴了很多博客,但是需要的东西都是散的,自己总结下吧,结果发现哇是真的累啊~~~