快速上手：在 Linux 上安装 PyTorch

引言

快速上手：在 Linux 上安装 PyTorch

PyTorch 是一个流行的深度学习框架，它为训练和部署机器学习模型提供了强大的工具。对于 Linux 用户来说，在系统上安装 PyTorch 非常简单。本文将提供逐步指南，帮助您在 Ubuntu 或 CentOS Linux 系统上快速安装 PyTorch。

前提条件

Linux 系统（Ubuntu 或 CentOS） Python 3.6 或更高版本 pip 包管理器

步骤 1：安装依赖项

``` sudo apt-get update sudo apt-get install python3-pip ```

步骤 2：安装 PyTorch

使用 pip 命令安装 PyTorch：

``` pip install torch torchvision ```

步骤 3：验证安装

要验证您的 PyTorch 安装是否成功，请在终端中运行以下命令：

``` python -c "import torch; print(torch.cuda.is_available())" ```

如果输出为 `True`，则表明您的系统已成功安装 PyTorch。

步骤 4：安装 CUDA（可选）

如果您需要使用 GPU 加速您的 PyTorch 模型，则需要安装 CUDA。请确保您的系统具备以下条件：

NVIDIA GPU CUDA 驱动程序

安装 CUDA 驱动程序后，运行以下命令以安装 PyTorch 的 CUDA 版本：

``` pip install torch torchvision torchtext -f https://download.pytorch.org/whl/cu100/torch_stable.html ```

使用 Anaconda 安装（可选）

如果您更喜欢使用 Anaconda 管理您的 Python 环境，则可以使用以下命令安装 PyTorch：

``` conda install -c pytorch pytorch torchvision torchaudio cudatoolkit=10.2 ```

卸载 PyTorch

要卸载 PyTorch，请运行以下命令：

``` sudo pip uninstall torch torchvision ```

结论