深度学习环境GPU配置

操作系统为 Windows10。 独显为英伟达显卡。

安装cuda

查看显卡是否支持cuda, 以及对应cuda版本。

打开NVIDIA控制面板, 左下角系统信息, 查看组件中cuda驱动版本。 安装的cuda不能高于此版本(可以低, 比如11.3.101 可以装cuda10.2)。

顺便

官网下载对应版本。

安装过程中注意选择自定义, 自定义组件中取消 VS。

C盘空间够的话尽量默认安在C盘, 记住默认路径。

安装完成后配置系统变量, 在path下添加

1
2
3
4
5
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\CUPTI\lib64
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.2\bin\win64
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.2\common\lib\x64

打开命令行输入 nvcc -V 检查是否安装成功。

安装cudnn

查询cuda对应的cudnn版本, 去官网下载, 需要注册登录以及填一些信息和问卷。

下载压缩包后解压, 将三个文件夹覆盖到cuda安装目录即可。(可以将文件夹的文件复制到cuda安装目录的对应文件夹中)。