cuda + cudnn + pytorch (win10)

win10 安装pytorch gpu 版

• [查 显卡驱动](#查 显卡驱动)
• [cuda 安装](#cuda 安装)
• [cudnn 安装](#cudnn 安装)
• 安装pytorch
• 测试

查 显卡驱动

• win + R : 输入cmd 唤出命令行；
• 输入nvidia-smi ；
  
• 关注驱动版本和cuda版本；
• 后面安装的cuda版本不高于当前的cuda版本；

cuda 安装

• 查看pytorch需要的cuda版本
  
• cuda官网下载
  
• 全部下一步自动安装。

cudnn 安装

• 进入官网下载对应cuda版本的cudnn
  
• 解压后，把解压出来的bin,include,lib文件夹内容，分别放入cuda对应的文件夹中。参考路径：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.6

安装pytorch

• 在anaconda 中创建pytorch环境。conda create -n myEnvName python=3.9
• 运行前面官网的安装指令。

测试