如何安装 PyTorch

2023-10-23 19:29:58

前言

现在使用 pytorch 框架进行深度学习的人越来越多了，但是 torch 的安装（GPU 版本）和普通第三方包的安装过程不太一样，因为涉及到电脑的显卡驱动版本，下面来看看安装 pytorch 的正确姿势。

打开 NVIDIA 控制面板，点击工具栏帮助 --> 系统信息，在弹出的对话框中有如下信息：

可以看到本机驱动程序版本为 466.92，接着前往 NVIDIA CUDA Toolkit Release Notes，在 cuda driver 这一节有下面这个表：

由于本机为 Windows 系统且驱动版本为 466.92，所以应该选择 CUDA 11.6.x。

在 PyTorch 官网的 Get Started 页面有下图所示的安装向导：

虽然没找到 CUDA 11.6.x，但是 CUDA 11.3 也是可以的，复制 Run this Command 里面的指令：

pip install torch==1.10.2+cu113 torchvision==0.11.3+cu113 torchaudio===0.10.2+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

这里把 pip3 换成了 pip，不过 pip3 也是可以的。在终端运行上述指令，等待安装完成即可。