CUDA 的安装
使用 NVIDIA GPU 编写程序需要安装 CUDA。
CUDA 的安装包括 CUDA Toolkit 和 CUDnn。
安装 CUDA Toolkit
首先，需要在自己的电脑上找到并打开 NVDIA 控制面板，点击左下角的【系统信息】，查看自己 GPU 支持的最高 CUDA 版本。
然后，在官方网站找到自己需要的 CUDA 版本。
这里由于我是 Win11 系统所以选择了 11.6 版本。
选择需要的安装版本并下载：
然后走流程安装就对了。
- 会提示两次选择路径，第一次选择的是临时文件下载路径，第二次才是安装路径，不要选择相同路径。
- 选择自定义安装并全选，可以取消 VS 相关插件的安装。
完成之后使用命令 nvcc -V 查看是否安装成功。成功截图如下：
安装 CUDnn
官网链接
- 需要注册一个账号，随便乱填就行，全选免费即可。
- 由于没有 CUDA 11.6 专用的 CUDnn 版本，可以直接选项 CUDA 11.x 版的。
下载后是一个 .zip， 解压后将 /bin /include /lib 复制到上一节选择的 CUDA 安装目录下的对应文件夹。
在 extras\demo_suite 下运行 bandwidthTest.exe 和 deviceQuery.exe 检查是否安装成功。
若最后有 Result = PASS 表示成功。