如何在 Ubuntu 上安装 PyTorch

简介

PyTorch 因其易用性、动态计算图和高效性而日益流行，成为实现深度学习模型的首选。如果你想探索这个工具并学习如何在 Ubuntu 上安装 PyTorch，本指南将对你有所帮助！

在本教程中，我们将引导你完成在 Ubuntu 系统上使用 Pip 或 Anaconda 安装 PyTorch 的过程，帮助你开始利用这个工具。

安装和配置步骤

第一步：更新系统软件包

首先，更新系统软件包，以确保你使用的是最新的软件包。

sudo apt update

第二步：安装 Python3-venv

接下来，你需要安装 Python3-venv，你可以使用它为你的项目创建一个隔离的 Python 环境，这样你就不必全局安装 Python 包，从而防止可能的兼容性问题。

sudo apt install python3-venv -y

第三步：设置环境

现在创建并导航到一个新的目录，该目录将用于存储虚拟环境文件和设置。这里 “pytorch_env” 是目录的名称；如果需要，你可以使用不同的名称。

mkdir pytorch_env
cd pytorch_env

接下来，运行以下命令来创建并激活虚拟环境。

python3 -m venv pytorch_env
source pytorch_env/bin/activate

一旦虚拟环境被激活，你应该看到你的命令提示符发生变化，显示虚拟环境的名称。激活此环境后，你可以安装与此虚拟环境相关的 Python 包或脚本，而不是系统范围的 Python，从而避免与需要不同包版本的项目发生冲突。

第四步：使用 Pip 安装 PyTorch

现在，你的虚拟环境已激活，你可以继续在 Ubuntu 系统上安装 PyTorch。

仅支持 CPU：

要仅安装支持 CPU 的 PyTorch，请运行：

pip3 install torch torchvision torchaudio --index-url <https://download.pytorch.org/whl/cpu>

支持 GPU (安装 CUDA 与 PyTorch) (CUDA)：

要安装支持 GPU 的 PyTorch，请运行：

pip3 install torch torchvision torchaudio

第五步：验证安装

你可以运行以下命令来验证 PyTorch 是否已安装：

python
import torch
print(torch.__version__)

上面的命令将启动 Python 解释器，导入 PyTorch 库并打印当前安装的 PyTorch 版本。

使用 Conda 安装 PyTorch

你还可以使用名为 Anaconda 的开源平台来安装 PyTorch。 你可以通过以下步骤使用 Conda 安装 PyTorch：

第一步：更新系统软件包

像往常一样，你需要确保你的系统软件包是最新的。

sudo apt update

第二步：安装 Anaconda

接下来，你需要安装 Anaconda，你可以按照以下步骤进行安装。

首先，你需要安装 Curl。这将用于下载 Anaconda 安装程序脚本。

sudo apt install curl -y

接下来，作为最佳实践，你需要隔离安装过程。导航到 “tmp” 目录，如果你需要，你可以创建或使用不同的目录。

cd /tmp

接下来，通过运行以下命令，使用 Curl 下载 Anaconda 安装程序脚本：

curl --output anaconda.sh https://repo.anaconda.com/archive/Anaconda3-2023.07-2-Linux-x86_64.sh

上面的命令将下载 Anaconda 安装程序脚本并将其保存为名为 “anaconda.sh” 的文件。你始终可以在他们的网站上获取最新版本脚本的 URL：https://repo.anaconda.com/archive。

现在使用以下命令验证下载文件的完整性：

sha256sum anaconda.sh

你需要确认 sha256sum 校验和值与官方网站上的值匹配：https://repo.anaconda.com/archive。

验证下载脚本后，你可以继续运行 Anaconda 安装程序脚本并通过运行以下命令开始安装过程：

bash anaconda.sh

安装过程完成后，你需要更新当前 shell 会话，以确保 Anaconda 环境及其命令可用于当前的终端会话。通过运行以下命令执行此操作：

source ~/.bashrc

它还会激活 Anaconda，正如你在提示符更改为 “base” 时所看到的那样，这是 Anaconda 在安装过程中创建的默认环境。 随后，你可以使用 conda activate 命令来激活 Anaconda。

你可以使用 conda --version 命令来验证 Anaconda 是否已安装。

第三步：使用 Anaconda 安装 PyTorch

现在，Anaconda 已安装并激活，你可以使用它来安装 PyTorch。

仅支持 CPU：

conda install pytorch torchvision torchaudio cpuonly -c pytorch

支持 GPU (安装带有 CUDA 的 PyTorch)：

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

第四步：验证安装

运行以下命令来验证 PyTorch 是否已安装：

python
import torch
print(torch.__version__)

如何卸载 PyTorch？

如果你需要在 Ubuntu 系统上卸载 PyTorch，你可以通过运行以下命令使用 Pip 来执行此操作：

pip3 uninstall torch -y

要使用 Anaconda 卸载 PyTorch，请运行：

conda remove pytorch -y

结尾

至此，你的系统上应该已经安装了 PyTorch。本指南展示了如何使用 Pip 和 Anaconda 在 Ubuntu 上安装 PyTorch。 你还了解到 PyTorch 提供了基于系统上可用计算资源下载的选项。 在解决了 PyTorch 的安装问题后，你可以继续探索其功能并体验其效率。

我的博客：https://blog.ivwv.site