深度学习
Windows下开发环境搭建
3. CUDA环境安装
参考链接:
官方文档:https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html
系统要求:
翻译:
要在你的系统上使用CUDA,需要安装:
- 支持CUDA的GPU
- 支持的Microsoft Windows版本
- 支持的Microsoft Visual Studio版本
- NVIDIA CUDA Toolkit(下载地址: http://developer.nvidia.com/cuda-downloads)
3.1 首先检查并更新一下自己的显卡驱动
官网CUDA工具包和兼容的驱动程序版本对照:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
-
鼠标右键,打开NVIDIA控制面板
-
点击左下角的系统信息
-
可以看到自己的显卡型号,我的这里是GeForce 940MX(记住这个型号就可以了),驱动程序版本还是388.57的,比较老了。
-
点击组件,看到显卡支持的CUDA还是9.1的
-
对照上面的表格,可以看到如果要装CUDA9.1的话,驱动程序建议为>=391.29。
-
上面的对照啥的只是看看,只需要知道自己的显卡型号就行,下面更新一些对应显卡的驱动
显卡驱动程序下载:https://www.nvidia.com/download/index.aspx?lang=en-us#
-
如下图,找到自己对应的显卡。即Product栏中有上面看到的显卡型号(我这里是GeForce 940MX)
-
点击SEARCH按钮,如果搜索出来没有结果,就修改一下
Download Type
或Language
,搜索到了的话,会出现下图,点击下载按钮。
-
继续点击下载按钮
-
开始下载。(如果浏览器下载比较慢的话,可以把下载链接复制到迅雷进行下载)
-
下载下来之后,双击程序,开始安装
-
可以更改一下安装路径(不过好像改了也没什么用,安装完后这个路径找不到了),点击OK
-
等待安装
-
选择同意并继续
-
选择自定义安装,点击下一步
-
点击下一步
-
等待安装
-
安装完成,点击关闭。
-
再去看一下自己的显卡驱动程序。可以看到,驱动程序版本已经从之前的388.57更新到了457.30,现在已经也可以支持CUDA 11.1了
-
OK,到这里显卡驱动就更新完了,下面来安装目前最新的CUDA 11.1
3.2 安装CUDA 11.1
官方文档:https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html
- 在http://developer.nvidia.com/cuda-downloads中选择目标平台,并下载cuda 11.1.0
-
选择位置,并保存,开始下载。(如果浏览器下载慢的话,可以把下载链接复制到迅雷中,使用迅雷下载会快很多)
-
下载下来之后,双击应用程序,开始安装
-
更改一下安装路径(不过好像没什么用,安装完之后cuda_11.1目录不见了),点击OK
-
等待安装
-
点击同意并继续
-
选择自定义,点击下一步
-
由于之前我们已经更新过驱动程序了,这里只用勾选第一个就可以了。点击下一步
-
这里的安装位置可以不用改,点击下一步
-
我犹豫了一下,我自己决定还是改一下安装位置。(等安装完了发现,这里的还是不改安装路径比较好,后面会有几个地方在官方文档中都是用的上图↑中的路径,如果这里改了路径的话,之后就要仔细对照着来,比较麻烦。 当然,我这里实际安装时是采用的修改之后的安装路径,即下图中的路径)。点击下一步
-
等待安装
-
安装过程中报了个错,点击确定即可,好像不影响。
-
安装完成,点击下一步
-
点击关闭
-
打开cmd命令行窗口,输入
nvcc -V
,回车,可以看到如下信息。
-
再看一下环境变量中的系统变量,可以看到,安装过程中自动帮我们添加了两个变量:
CUDA_PATH
和CUDA_PATH_V11_1
。
-
验证安装是否成功
如果上面那里没有更改路径的话,可以参考官方文档(https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html#verify-installation)中的路径,来找到
deviceQuery
和bandwidthTest
-
如果更改了路径的话,对照自己更改的路径,
deviceQuery
和bandwidthTest
都在extras\demo_suite
目录下。在...\extras\demo_suite
的目录下打开cmd命令行窗口,分别输入.\deviceQuery.exe
和.\bandwidthTest.exe
,如下图,如果Result均为PASS,则说明CUDA安装成功
-
OK,到这里CUDA就安装好了。
3.3 安装cuDNN
官方文档:https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#install-windows
下载地址:https://developer.nvidia.com/rdp/cudnn-download(需要先登录)
-
在下载cuDNN之前,需要先注册一个账号并登录。
-
在https://developer.nvidia.com/developer-program中先注册一个账号,并登录。在https://developer.nvidia.com/rdp/cudnn-download中点击下载。勾选I Agree后,选择for CUDA 11.1的那个,
-
再选择Windows版的下载。(如果浏览器下载比较慢的话,可以把下载链接复制到迅雷中,使用迅雷下载会快很多)
-
下载下来是一个压缩包,把压缩包解压到当前目录。
-
如果之前安装CUDA时没有更改过路径,那就可以安装官方文档中的路径,将文件复制粘贴过去。
-
如果之前安装CUDA时更改过路径的话,就把cuda下面的
bin
、include
、lib
这三个文件夹复制粘贴到CUDA Development
(对应之前更改后的安装路径)下。
- OK,粘贴过来之后就可以了。到这里cuDNN就安装好了。
创作不易,喜欢的话加个关注点个赞,❤谢谢谢谢❤