如何在 AlmaLinux 上安装 PyTorch
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：
如果使用 CUDA，请确保 NVIDIA 驱动程序和 CUDA toolkit 已正确安装，并且与您正在安装的 PyTorch 版本兼容。
结论
您已成功在 AlmaLinux 上安装 PyTorch。无论您是在开发 deep learning 模型还是运行实验，PyTorch 都为 Linux 系统上的 machine learning 提供了一个灵活而强大的框架。对于更高级的设置，可以考虑与 Jupyter Notebook 集成或使用 Docker 容器。
相关文章
