前置工作

  • 升级显卡驱动
  • 安装Visual Studio的桌面应用开发套件(选做？似乎不影响PyTorch)

安装CUDA

  • PyTorch可能需要某个指定版本的CUDA，从官网的归档里可以下载到特定版本的安装包
  • 一路下一步直到选择组件的步骤
  • 选择组件时留意一下CUDA以外的组件版本号，安装包附带的其他组件可能并不是最新的，此种情况下则不覆盖安装旧版本其他组件了
  • 勾选上CUDA，下一步

现在的CUDA安装包会自动配置好PATH，等安装完成后就基本没有需要手动操作的地方了。

验证安装

命令行执行以下命令查看安装的CUDA版本：

nvcc --version

然后前往C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\extras\demo_suite\(默认安装路径)，在命令行运行deviceQuery.exe和bandwidthTest.exe。若测试结果均为PASS，则CUDA环境已经成功安装。