1、在NVIDIA控制面板查找本机对应CUDA版本
-----如图,本机型适配CUDA版本为12.4
进入官网下载:kCUDA Toolkit Archive | NVIDIA Developer
2、下载cudnn
进入官网下载:cuDNN Archive | NVIDIA Developer 查找与自己cuda版本适配的cudnn
笔者在此便选取v8.97即可
实际上下载的是一个压缩包,把其同名文件夹的内容复制到cuda对应文件夹中即可
3、环境配置
点击设置-->搜索高级系统设置-->查看环境变量
环境变量要配置以下四个(系统可能会漏部分,自己补就好了,注意自己的安装路径)
验证是否安装成功:
运行cmd,输入nvcc --version 即可查看版本号;
set cuda,可以查看 CUDA 设置的环境变量。 下图是成功安装的情况
差不多就是这些了。还是安装的比较顺利的~