1. CUDA安装
Nvidia官网下载对应版本CUDA Toolkit
CUDA Toolkit 12.1 Downloads | NVIDIA Developer
CUDA Toolkit 12.5 Downloads | NVIDIA Developer
安装配置步骤参考:配置显卡cuda与配置pytorch - 知乎 (zhihu.com)
2. 根据CUDA版本,安装cudnn
2.1下载:
cuDNN Archive | NVIDIA Developer
2.2 安装:
对下载的cuDNN压缩包解压后出现如下三个文件夹子,
3.3为cuDNN添加环境变量:
找到环境变量-系统变量-path,分别将如下三个变量添加进去,完成安装