在win10系统下配置cuda11.2+cudnn8.1.0,电脑显卡驱动版本:551.61(打开终端通过nvidia-smi
查看对应驱动信息,其中右上角“CUDA Version”表示的最高支持的cuda版本,大于11.2即可)
一、cuda11.2配置
1、下载及安装
(1)通过cuda官方下载地址根据自己电脑配置找到对应版本的安装包下载,例如我是win10系统,选择了如下版本安装包:
(2)双击exe文件安装,解压缩地址按需选,一般默认即可
(3)根据提示点击【下一步】继续安装
- 选择自定义安装
- 电脑本身已经装的适合的显卡驱动的话,这里就不需要再装了,只勾选【CUDA】即可
- 安装位置可以更改,也可以就按照它默认安装位置,如果更改了,一定要记住更改后安装位置,后续配置环境变量可能会用到
- 等待安装完成即可
2、测试
通过cmd打开新的终端,输入指令nvcc -V
查看cuda版本,显示和安装版本一致的信息则说明安装成功,例如:
3、如果输出版本信息不一致请看本步骤,否则忽略即可
版本信息不一致,最大的可能是因为电脑上之前装过其他版本的cuda。
例如,我的电脑上之前装了cuda10.2,按照上一步安装完成后通过nvcc -V
命令测试出来的不是新安装的11.2版本,而是10.2版本,此时需要配置环境变量信息
- 删除和修改系统环境变量10.2对应信息
- 注意:删除之前确认一下是否有正确版本的环境变量信息在,保证有正确的版本路径在的情况下把旧版本对应环境变量信息删除
下述步骤5的两条环境路径信息我的电脑里没有对应11.2的,确认自己的安装位置对应路径存在后修改10.2为11.2
- 删除/修改完毕后,重新使用
nvcc -V
测试,版本显示为刚才新装的11.2则成功,至此cuda安装完毕!
二、cudnn配置
(1)通过cudnn官网下载与cuda、电脑系统、架构对应版本的cudnn安装包,例如,我下载的版本为
- 注意:下载前需注册NVIDIA账号
(2)解压下载好的cudnn压缩包
(3)将解压后的bin、include、lib文件夹复制至cuda安装路径下对应文件夹即可。