如何在 Ubuntu 22.04 上安装 PyTorch？（PyTorch Ubuntu 分步安装指南）
PyTorch 是一款广泛使用的开源机器学习框架，以易用性、动态计算图和高效率著称，非常适合深度学习任务。
PyTorch 由 Meta AI 开发，现已成为 Linux 基金会的一部分，被用于图像和语音识别、自然语言处理以及强化学习等应用。其强大且针对 GPU 优化的架构让数据科学家能够轻松构建和部署神经网络，是 AI 研究与开发中必不可少的工具。
本指南将帮助您使用 Pip 或 Anaconda 在 Ubuntu 上安装 PyTorch，快速上手。
前提条件
请确保您的系统满足以下要求：
- Python 3.6 或最新版本
- 如需 GPU 支持，请确保已安装 CUDA 驱动程序和工具包。
在 Ubuntu 上安装 PyTorch（Ubuntu 安装 PyTorch）
要在 Ubuntu 系统上安装支持 CPU 或 GPU 的 PyTorch，可以使用 Pip 或 Anaconda。我们将重点介绍如何使用 Pip 安装 PyTorch，以及如何设置虚拟环境来有序管理 Python 软件包。
方法一：使用 Pip3 在 Ubuntu 上安装 PyTorch
按照以下步骤，您可以使用 Pip（Python 包管理器）在 Ubuntu 上轻松安装 PyTorch：
第 1 步：更新系统软件包
首先，确保系统软件包是最新的。打开终端并运行以下命令：
$ sudo apt update
该命令会更新软件包列表，确保您获取的是最新版本。
第 2 步：安装 Python3-venv
接下来，需要安装 python3-venv，它可以用来创建相互隔离的 Python 环境。这一点很重要，可以避免需要不同软件包版本的项目之间发生冲突。
$ sudo apt install python3-venv -y
第 3 步：设置 Python 虚拟环境
要设置 Python 虚拟环境，请为您的 Python 项目创建一个目录，并使用 cd 命令进入该目录。该目录将用于存放虚拟环境及相关文件。例如，我们创建了一个名为“pytorch_env”的目录。
$ mkdir pytorch_env
$ cd pytorch_env
现在，创建并激活虚拟环境：
python3 -m venv pytorch_env
source pytorch_env/bin/activate
激活虚拟环境后，终端提示符会发生变化，表明您现在正在 pytorch_env 环境中工作。此时，您可以专门在该环境中安装 Python 软件包或运行脚本。这样可以将依赖项与系统级 Python 安装分开，避免与其他项目发生冲突。
第 4 步：使用 Pip 安装 PyTorch
在虚拟环境处于激活状态时，就可以安装 PyTorch 了。如果您没有 GPU 或不需要 GPU 加速，可以安装仅支持 CPU 的 PyTorch 库。要安装在 CPU 上运行的 PyTorch，请使用以下命令：
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
如果您拥有兼容的 GPU 并希望利用 CUDA 进行加速，请安装支持 GPU 的 PyTorch：
pip3 install torch torchvision torchaudio
第 5 步：验证安装
要确认 PyTorch 已成功安装，请打开 Python 解释器并查看 PyTorch 版本：
import torch
print(torch.__version__)
这将显示您安装的 PyTorch 版本，证明安装成功。
方法二：如何使用 Anaconda 在 Ubuntu 上安装 PyTorch
您还可以使用流行的开源平台 Anaconda 在 Ubuntu 上安装 PyTorch。这种方法可以让您轻松管理 Python 软件包和环境。要使用 Conda 安装 PyTorch，请按以下步骤操作：
第 1 步：更新系统软件包
首先更新系统软件包，确保一切都是最新的。打开终端并运行：
$ sudo apt update
第 2 步：安装 Anaconda 或 Conda
接下来，需要安装 Anaconda。您可以使用 Curl 命令下载 Conda 安装包。如果您的 Ubuntu 系统尚未安装 curl，请使用以下命令安装：
$ 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”。如有需要，您可以在 Anaconda 官方网站上查看最新版本。
验证下载的文件。检查下载文件的完整性非常重要，可以确保文件没有损坏。使用以下命令进行验证：
sha256sum anaconda.sh
请确保校验和与 Anaconda 官方网站上提供的值一致。
验证完成后，运行安装脚本进行安装：
bash anaconda.sh
安装完成后，刷新 shell 会话，使 Anaconda 命令在终端中可用：
source ~/.bashrc
您会看到终端提示符变为“base”，表示默认的 Anaconda 环境已激活。之后，您可以使用 conda activate 命令激活 Anaconda 环境。
运行以下命令确认 Anaconda 已安装：
conda --version
第 3 步：使用 Anaconda 安装 PyTorch
安装并激活 Anaconda 后，就可以安装 PyTorch 了。
如果您的系统没有 GPU 或不需要 GPU 加速，请安装支持 CPU 的 PyTorch：
conda install pytorch torchvision torchaudio cpuonly -c pytorch
如果您拥有 GPU 并希望使用 CUDA 进行加速，请安装支持 GPU 的 PyTorch：
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
第 4 步：验证安装
最后，要验证 PyTorch 是否安装正确，请启动 Python 会话并查看 PyTorch 版本：
import torch
print(torch.__version__)
这将输出当前安装的 PyTorch 版本，证明一切都已正确设置。
如何在 Ubuntu 上卸载 PyTorch？
要从 Ubuntu 系统中卸载 PyTorch，请根据最初的安装方式按以下步骤操作。
如果您是使用 Pip 安装的 PyTorch，可以运行以下命令将其删除：
$ pip3 uninstall torch -y
该命令会从您的 Python 环境中卸载 PyTorch。
如果您是使用 Anaconda 安装的 PyTorch，可以使用以下命令卸载：
$ conda remove pytorch -y
这将从您的 Anaconda 环境中删除 PyTorch。
结论
本指南演示了如何使用 Pip 和 Anaconda 在 Ubuntu 上安装 PyTorch，还介绍了根据系统计算资源可选的不同安装方式。现在 PyTorch 已经安装完毕，您可以开始探索它的各项功能，充分发挥其强大能力。
VPS 主机是一种强大而灵活的选择，能够显著影响网站的成功。借助 BlueVPS，您可以租用遍布全球的服务器，并可使用 cPanel 轻松管理。准备好提升网站性能了吗？现在就考虑 VPS 主机吧。
blog