根据自己的显卡型号、显卡驱动版本,安装对应的cuda与cudnn。
查看Cuda版本
打开cmd,输入nvcc --version
查看Cudnn版本
进入cuda的安装目录中
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1\include
用Visual Studio打开cudnn_version.h文件
Cudnn版本为8.8.1
根据自己的显卡型号、显卡驱动版本,安装对应的cuda与cudnn。
查看Cuda版本
打开cmd,输入nvcc --version
查看Cudnn版本
进入cuda的安装目录中
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1\include
用Visual Studio打开cudnn_version.h文件
Cudnn版本为8.8.1