windows安装gpu版pytoch全过程

显卡驱动

到这里下载显卡驱动更新工具，该工具负责安装最新的驱动及更新

查看可以安装cuda版本

nvidia-smi

例如，我这里可以安装的cuda版本为：12.1

下载cuda并安装

下载

点击这里进入下载页面：

这里下载 12.1 大版本即可

安装

取消这个选项：

安装路径看情况是否使用默认路径，只要以下环境变量配置正确：

安装cudnn

• 点击这里下载
• 将压缩包解压，然后把包含的3个文件夹复制到CUDA安装路径下即可

最新：其实安装pytorch-gpu版，不需要单独安装上面的CUDA和cuDNN，torch的安装包集成的有这俩玩意，你只需要保证你的N卡的驱动版本号高于torch的最低要求。直接参考下面安装pytorch

安装pytorch

进入官网，在你的解释器运行下图红框的命令：

验证

import torch
torch.cuda.is_available()