CUDA的安装需要两个包:CUDA Toolkit
CUDNN(算是CUDA的一个补丁,用于优化深度学习)
CUDA下载官方教程:Installation Guide Windows :: CUDA Toolkit Documentation
CUDNN下载官方教程:Installation Guide :: NVIDIA Deep Learning cuDNN Documentation
一般先安装CUDA,再安装CUDNN
搜索栏搜“NVIDIA"并点击进入NVIDIA控制面板
点击左下角的“系统信息”
查看组件部分的信息,如图中表示兼容11.6.1的CUDA
下载网址:CUDA toolkit Download
题主根据显卡兼容信息选择的CUDA Toolkit 11.6.1的链接下载(点前面一半的链接)
选择“Windows”→“x86_64”→“11”→“exe_{local}”→“Download(2.5GB)”
【注意】要在准备安装的盘里面设置两个文件夹:空的解压文件夹(解压后这个文件夹会自动删除,不能将文件安装在里面),安装文件夹(包含两个小文件夹:Samples和v11.6,当然也可以自己再设置别的)
点击之前下载的cuda的.exe文件,将解压文件夹目录输入
等待检查系统兼容性
点击“同意并继续(A)”
这里推荐首次安装的小白选择“自定义(C)(高级)”,多次安装的选择“精简(E)(推荐)”
其中要注意“Visual Studio Integration”我没有选择,我已经安装了,大家自己选择
接下来就是“选择安装位置”,之前在安装文件夹设置了两个子文件夹,可以用到了,注意选择“Samples”文件只选择一个,其余的都放在“v11.6”当中(根据自己命名的来)
然后等它自己安装好
出现以下画面,点击“下一步”就可以了
搜索“高级系统设置”,点击“查看高级系统设置”
查看”高级“→”环境变量(N)“,在”环境变量“中的”系统变量“中 一般会自动生成”CUDA_PATH“和”CUDA_PATH_V11_6“两个系统变量,另外两个系统变量”NVCUDASAMPLES_ROOT“和”NVCUDASAMPLES11_6_ROOT“,需要手动添加
变量:CUDA_PATH 值:D:\CUDA\v11.6(按照自己安装路径)
变量:CUDA_PATH_V11_6 值:D:\CUDA\v11.6(按照自己安装路径)
变量:NVCUDASAMPLES_ROOT 值:D:\CUDA\Samples(按照Samples安装路径)
变量:NVCUDASAMPLES11_6_ROOT 值:D:\CUDA\Samples(按照Samples安装路径)
键盘“Win+R”,输入“cmd”
输入“nvcc -V” 或 “nvcc --version”查看安装CUDA的版本号
nvcc -V 查看版本号
nvcc --version 查看版本号
set cuda 查看环境变量
出现以上画面则代表CUDA安装成功
CUDNN安装网址:CUDNN-Download
首先就是需要你注册一个账号,可以注册,没别的影响,输入自己邮箱就可以,会给你的邮箱发一个验证的邮件,点开验证一下就可以登陆啦!
登陆后需要同意以后才可以选择下载的CUDNN
选择"Archived cuDNN Releases"可以查看所有版本的CUDNN
本人选择最上面的8.6.0的,按照自己CUDA对应的版本选择就好
下载后是个压缩文件,在下载目录下解压就好,可以删除的
解压后我们需要复制其中的三个子文件夹“bin” “include” “lib”
直接复制到之前设置的CUDA文件中的子文件夹“v11.6”中,主要将同样的三个文件夹覆盖,
还是打开“查看高级系统设置”,点击系统变量中的“Path”→“编辑”→“新建”,按照自己路径文件夹来配置就好,添加以下四个路径
D:\CUDA\v11.6\bin
D:\CUDA\v11.6\include
D:\CUDA\v11.6\lib
D:\CUDA\v11.6\libnvvp
在“D:\CUDA\v11.6\extras\demo_suite”文件夹中启动终端
分别运行“deviceQuery.exe”和“bandwidthTest.exe”两个程序,输入以下:
.\deviceQuery.exe
.\bandwidthTest.exe
综上就算安装完成CUDA,接下来大家就可以根据需要配置自己的深度学习环境了哦~!