PC端配置深度学习环境本身不太麻烦,但是如果想要使用GPU加速,那么就有点麻烦了。
一是因为CUDA以及CUDNN的下载很麻烦,因为服务器在国外;二是因为各种版本的匹配问题,因为各种版本匹配的涉及到硬件软件深度学习框架厂家三方。
本文记载一下安装过程,顺便提供CUDA以及CUDNN的安装包,见百度云链接。
软件硬件背景:
OS:Windows 10
GPPU:GeForce GTX 1660Ti
1 查看自己的显卡版本
以英伟达显卡为例
打开英伟达控制面板,点击帮助,在下拉菜单选择系统信息,如下图所示:
然后在新的界面点击组件,这时候你可以看到你的显卡相关信息了。着重留意你的CUDA版本信息,在本列中,版本为11.1,那么在下载CUDA安装包的时候,版本不得高于11.1。
2 下载CUDA以及CUDNN库
2.1 CUDA的下载
CUDA的下载链接如下:
https://developer.nvidia.com/zh-cn/cuda-downloads
在你进入之后的界面如下:
他的默认选择是11.1版本的包下载界面,这时候,我建议你不要使用最新版本的包,因为各个框架的厂家对最新的支持没那么快,建议采用低版本的安装,比如最新版本是11.x,那么我建议你优先考虑10.X,这样有一个好处就是能确保各个框架都能支持。更保险的做法是去你要使用的框架的主页去查看他们支持哪些版本,以pytorch为例,如下图:
首界面提示的是可以支持10.2和11.1,那么你在安装的时候,就可以考虑安装这两个版本。如果你的机器不支持这两个版本,那么你可以点击左下角Previous version of Pytorch去查看她的往期版本,寻找合适的版本下载和安装。
回到前面说的主界面只有cuda11的下载链接,那么我们就需要去寻找其他版本的下载链接,在屏幕的下方可以看到一个早起版本的字眼,点击他:
进去之后的界面如下:
选上图选择符合你需求的版本点击进去,之后显示画面如下:
然后再继续选择和你系统匹配的版本,点击下载,如上图所示。
2.2 CUDNN的下载
CUDNN的下载链接如下:
https://developer.nvidia.com/rdp/cudnn-download
进入之后,你需要注册一个英伟达账号,然后回答一系列问题,才能够正常下载,此时的下载界面如下:
此时主界面只有两个版本的CUDNN可以下载,如果主界面的两个版本与你的CUDA版本不匹配,那么就点击Archived cuDNN Releases,这里会有所有的历史版本链接
然后去界面选择要下载的CUDNN版本,在这里要注意,一定要下载与CUDA相匹配的CUDNN版本!!!
3 CUDA以及CUDNN安装
3.1 CUDA安装
双击下载好的CUDA,如果C盘足够大,建议使用默认路径,可以减少不必要的麻烦。如果C盘不够用,可以自定义其他地方为安装路径。不论安装在哪里,都要记住这个路径,因为后面会用到。
3.2 CUDNN安装
将下载的CUDNN文件解压出来,你会看到如下几个文件
此时,将这三个文件夹内的内容,分别复制到CUDA安装根目录对应的文件夹内,注意,是复制文件,不是直接复制文件夹。
如果你使用的是默认安装目录,那么你的目录应该是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA。在这个目录下,也存在bin,include,lib这三个文件夹
3.3 添加环境变量
右键单击‘此电脑’然后选择‘更多’,然后点击‘属性’
然后选择高级系统设置
然后点击‘环境变量’
然后,选择‘path’,然后点击编辑:
然后在新出来的界面点击‘新建’
将如下几个路径添加到环境变量中:
1. C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin
2. C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\libnvvp
3. C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin
4. C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\CUPTI\lib64
注意:要填写你自己的安装路径!!!
3.4 测试是否安装成功
3.4.1 进入Windows命令行
3.4.2 进入 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\demo_suite目录
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\demo_suite
运行deviceQuery.exe和bandwidthTest.exe文件
如果都没有报错,那么就是安装成功了。
你还可以在cmd下输入nvcc -V查看CUDA的版本
至此,显卡的硬件配置完全完成了。