PyTorch 是一个流行的开源机器学习库，适用于深度学习。在 Linux 系统上安装 PyTorch 非常简单，以下是一些详细的步骤。

系统要求

在开始安装之前，请确保您的 Linux 系统满足以下要求：

  • 操作系统：Ubuntu 16.04/18.04/20.04 或 CentOS 7/8
  • Python 版本：Python 3.6 - 3.8
  • 编译器：GCC 5.3 或更高版本

安装步骤

  1. 更新包列表：
sudo apt-get update
  1. 安装依赖项：
sudo apt-get install -y python3-pip python3-dev
  1. 安装 PyTorch：

首先，您需要确定要安装的 PyTorch 版本和是否需要 GPU 支持。以下是一些选项：

  • CPU 版本：
pip3 install torch torchvision torchaudio
  • GPU 版本：
pip3 install torch torchvision torchaudio cpuonly
  1. 验证安装：

安装完成后，您可以通过以下命令验证 PyTorch 是否已正确安装：

python3 -c "import torch; print(torch.__version__)"

额外资源

如果您需要更多关于 PyTorch 安装的帮助，可以访问我们的PyTorch 安装指南。

相关教程

如果您想要进一步学习 PyTorch，可以查看我们的PyTorch 教程。

PyTorch Logo