win10 安装pytorch gpu 版
查 显卡驱动
- win + R : 输入cmd 唤出命令行；
- 输入nvidia-smi ；
- 关注驱动版本和cuda版本；
- 后面安装的cuda版本不高于当前的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
- 运行前面官网的安装指令。
测试
