1.前置条件
安装了cuda和cudnn,其中cudnn是压缩版
1、安装cuda和cuDNN
首先系统支持的cuda,打开nvidia控制面板,点击左下方的“系统信息”,点击“组件”,显示版本是“12.1.112”。
到cuda官网下载相应版本,一般是向下兼容(低一点的版本是可以的)
kCUDA Toolkit Archive | NVIDIA Developer
网上很多下载及安装教程,这里不累赘。但是有个tips:第一次安装,以下页面建议选择“自定义”选项(借用大神的图)
自定义选项里不要勾选“Visual Studio”(前面步骤没有装的话,这又是一个tips:有人建议先到微软官网安装Visual Studio,但是后来发现不安装也能用)
安装完成后,大部分需要手动添加环境变量,方便检测是否安装成功。
win11是在桌面右键,点击“显示设置”,点击左边选择“系统”选项,选择右边的“系统信息”选项,“高级系统设置”-“高级”-“环境变量”-“系统变量”-”path”-“编辑”-“新建”,把相关内容加进去。(一共四个环境变量,一般是自动生成2个cuda_path和cuda_path_v12.1,另外两个手动添加)
至此&#