如何在 AlmaLinux 上安装 PyTorch

热门:
提升您的服务器设置! 申请 AVA 并使用 立减 15%
使用优惠码:

PyTorch 是由 Facebook 的 AI Research 实验室开发的最受欢迎的开源 machine learning 框架之一。它广泛用于 deep learning 应用,包括 computer vision 和 natural language processing。如果您使用的是 AlmaLinux,这是一款稳定可靠的 基于 RHEL 的 Linux 发行版,本指南将向您展示如何快速高效地安装 PyTorch。

前提条件

在开始之前,请确保:

  • 您有一个正在运行的 AlmaLinux 8 或 9 实例

  • 您拥有 root 或 sudo 权限

  • 已安装 Python 3.7 或更高版本

步骤 1:更新您的系统

打开您的终端并运行以下命令以更新所有系统软件包:

sudo dnf update -y

可选地,安装 EPEL 以访问更多软件包:

sudo dnf install epel-release -y

步骤 2:安装 Python 和 pip

检查 Python 是否已安装:

python3 --version

如果尚未安装,您可以使用以下命令安装:

sudo dnf install python3 python3-pip -y

步骤 3:创建虚拟环境(可选)

为了保持您的 Python 环境干净且隔离,建议创建一个虚拟环境:

python3 -m venv pytorch_env
source pytorch_env/bin/activate

步骤 4:使用 pip 安装 PyTorch

要安装仅支持 CPU 的 PyTorch:

pip install torch torchvision torchaudio

如果您想要 GPU 加速,并且拥有受支持的 NVIDIA GPU 以及正确安装的驱动程序和 CUDA,请使用相应的命令。例如,要安装支持 CUDA 11.8 的 PyTorch:

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

您可以在官方 PyTorch 网站上找到最新的安装选项:
https://pytorch.org/get-started/locally

步骤 5:验证安装

要验证 PyTorch 是否已正确安装,请运行:

python

然后在 Python shell 中输入以下内容:

import torch
print(torch.__version__)
print("CUDA available:", torch.cuda.is_available())

如果一切正常,您将看到已安装的 PyTorch 版本以及 CUDA 是否可用。

故障排除提示

  • 如果您在使用 pip 或 SSL 证书时遇到问题,请更新 pip:

pip install --upgrade pip
  • 如果使用 CUDA,请确保 NVIDIA 驱动程序和 CUDA toolkit 已正确安装,并且与您正在安装的 PyTorch 版本兼容。

结论

您已成功在 AlmaLinux 上安装 PyTorch。无论您是在开发 deep learning 模型还是运行实验,PyTorch 都为 Linux 系统上的 machine learning 提供了一个灵活而强大的框架。对于更高级的设置,可以考虑与 Jupyter Notebook 集成或使用 Docker 容器。

AvaHost Black Friday 全场服务 30% 优惠 工程师 24/7 支持 自有数据中心 · Chișinău DDoS 防护
AvaHost Black Friday 全场服务 30% 优惠 工程师 24/7 支持 自有数据中心 · Chișinău DDoS 防护
AvaHost Black Friday 全场服务 30% 优惠 工程师 24/7 支持 自有数据中心 · Chișinău DDoS 防护
AvaHost Black Friday 全场服务 30% 优惠 工程师 24/7 支持 自有数据中心 · Chișinău DDoS 防护