在 Linux 上安装 PyTorch

PyTorch是一个基于Python的科学计算包，它有以下特点：

• 具有强大的GPU加速
• 提供了灵活高效的深度学习模块
• 可以轻松移植到不同的平台

本文将向您介绍如何在Linux上安装PyTorch。

步骤1：安装Python

首先，我们需要安装Python。在Linux上，您可以使用以下命令安装Python：

sudo apt-get update
sudo apt-get install python3.8

步骤2：安装PyTorch的依赖项

在安装PyTorch之前，需要先安装它的依赖项。可以使用以下命令来安装这些依赖项：

sudo apt-get update
sudo apt-get install -y --no-install-recommends \
build-essential \
python3-dev \
python3-pip \
libjpeg-dev \
zlib1g-dev \
libpng-dev \
libvpx-dev \
libx264-dev \
libavcodec-dev \
libavformat-dev \
libswscale-dev \
libtbb2 \
libtbb-dev \
libjpeg8-dev \
libjasper-dev \
libdc1394-22-dev \
libcairo2-dev \
libopenexr-dev \
libgles2-mesa-dev \
libgbm-dev \
libgstreamer1.0-dev \
libgstreamer-plugins-base1.0-dev \
libavresample-dev \
libhdf5-dev \
libgflags-dev \
libgoogle-glog-dev \
liblmdb-dev \
libprotobuf-dev \
protobuf-compiler \
libsnappy-dev \
wget \
curl \
git

步骤3：安装PyTorch

现在，我们可以使用pip来安装PyTorch。您可以使用以下命令来安装PyTorch：

pip3 install torch torchvision torchaudio

如果您需要安装特定版本的PyTorch，请使用以下命令：

pip3 install torch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0

步骤4：验证安装

现在，我们已经成功地安装了PyTorch。让我们验证安装。

在Python中，输入以下命令：

import torch
print(torch.__version__)

如果上述命令的输出为PyTorch的版本号，则表明您已成功安装PyTorch。

结论

在本文中，我们介绍了在Linux上安装PyTorch的步骤。请按照上述步骤操作，您将成功地安装PyTorch，并可以开始使用它。