配置VSCode+Anaconda的PyTorch GPU环境笔记

真就是翻遍了整个互联网才终于让PyTorch在我的电脑上支持CUDA了。

叠个甲,我是纯菜鸟,说的不对你也别喷,喷了就是你对。

网上的教程都太有年代感了,跟着配置就是到处报错,最后配置出来的也是老版本,用着十分膈应(我个人不喜欢老版本的东西),所以想着记录一下我踩过的坑,让大家以后少走点弯路


下面直接进入正题:

在此之前,你需要自行完成:

  • 安装VSCode以及相关插件(汉化,Code Runner, Python包这三个就够了)
  •  安装Anaconda,掌握如何使用Anaconda创建虚拟环境
  • 掌握基本的Python使用方法
  • 掌握基本的命令行和终端使用方法
  • 安装自己的NVIDIA显卡,并且可以在驱动上正常运行

1.检查GPU

win + R打开命令行输入cmd并进入,

输入nvidia-smi检查你的gpu

配置VSCode+Anaconda的PyTorch GPU环境笔记_第1张图片

可以看到驱动版本(Driver Version)是552.12,CUDA最高支持版本(CUDA Version)是12.4

2.安装CUDA和cuDNN

去官网对照你的驱动版本以及CUDA版本下载支持的版本即可

你可能感兴趣的:(vscode,pytorch,笔记,conda,python)