ArchLinux下安装Pytorch

安装Pytorch

在 Arch Linux 下安装 PyTorch 的步骤如下：

1. 安装 Python

确保你的系统上已经安装了 Python。你可以使用以下命令来安装 Python 和 pip（Python 包管理工具）：

1

sudo pacman -S python python-pip

2. 安装 PyTorch

在 Arch Linux 上，可以通过 pip 安装 PyTorch。

1

pip3 install torch torchvision torchaudio

conda

pytorch

1

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

mamba

1

mamba install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia 

上面的命令将安装 PyTorch 及其相关的库 torchvision（计算机视觉相关工具）和 torchaudio（音频处理工具）。

3. 使用 AUR 安装（可选）

如果你更喜欢使用 Arch User Repository (AUR) 安装 PyTorch，可以使用以下步骤：

首先，你需要安装一个 AUR 助手，如 yay 或 paru。如果你还没有安装 AUR 助手，可以使用以下命令安装 yay：

1
2
3
4

sudo pacman -S --needed base-devel git
git clone https://aur.archlinux.org/yay.git
cd yay
makepkg -si

然后使用 yay 安装 PyTorch：

1

yay -S python-pytorch-cuda

4. 验证安装

安装完成后，你可以通过以下 Python 命令来验证 PyTorch 是否安装成功：

1
2

import torch
print(torch.__version__)

如果正确安装，你将看到 PyTorch 的版本号输出。

5. 安装 CUDA（如果需要）

如果你计划在 GPU 上使用 PyTorch 并且你的 GPU 支持 CUDA，你需要确保 CUDA 已经安装并正确配置。你可以通过安装 cuda 包来完成：

1

sudo pacman -S cuda cudnn

完成后，确保 CUDA 工具链路径已添加到你的 PATH 环境变量中：

1
2

export PATH=/opt/cuda/bin:$PATH
export LD_LIBRARY_PATH=/opt/cuda/lib64:$LD_LIBRARY_PATH

完成以上步骤后，你的 Arch Linux 系统应该已经成功安装了 PyTorch 并准备就绪。