cuda 安装#
1. cuda 的安装#
到 https://developer.nvidia.com/cuda-toolkit 去下载。在安装的时候一定要自定义安装,否则将会安装很多无用的东西。安装的选项,可以选择不更新驱动程序。
或者下载离线文件安装
安装,选择自定义安装。
安装后,和英伟达 cuda 相关的程序如下图所示。
注意,千万不要勾选 Nsight Visual Studio Edition 2019.2 等类似的无用的东西。
2. 测试环境是否安装成功#
运行 cmd,输入nvcc --version
即可查看版本号;
set cuda
,可以查看 cuda 设置的环境变量。
安装 cudnn#
cuDNN 地址如下,不过要注意的是,我们需要注册一个账号,才可以进入到下载界面。大家可以放心注册的。
cuDNN Download | NVIDIA Developer
下载结果是一个压缩包
cuDNN 配置#
cuDNN 叫配置更为准确,我们先把下载的 cuDNN 解压缩,会得到下面的文件:
解压后的文件:
下载后发现其实 cudnn 不是一个 exe 文件,而是一个压缩包,解压后,有三个文件夹,把三个文件夹拷贝到 cuda 的安装目录下。
CUDA 的安装路径在前面截图中有,或者打开电脑的环境变量查看,默认的安装路径如下:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\你安装的cuda版本号
拷贝时看到,CUDA 的安装目录中,有和 cuDNN 解压缩后的同名文件夹,这里注意,不需要担心,直接复制即可。cuDNN 解压缩后的同名文件夹中的配置文件会添加到 CUDA 安装目录中的同名文件夹中。【此处还是建议还是分别把文件夹的内容复制到对应文件夹中去】
cuDNN 其实是 CUDA 的一个补丁,专为深度学习运算进行优化的。然后再添加环境变量
首先把 cudnn 的同名文件复制到安装 cuda 的位置
配置环境变量
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\libnvvp