最详细的ARM架构服务器上配置Pytorch的教程!!!

1 阅读2分钟

准备系统与Python环境

  • 看服务器的架构

输入下面指令进行查看架构

uname -m

当前我使用的服务器架构为aarch64是基于ARM架构的,不同于传统的x86_64架构

  • 安装miniconda(ARM64版本)

由于标准的 Anaconda/Miniconda 默认是 x86 架构,你需要下载对应的 AArch64 版本:

Bash

# 下载 Miniconda ARM64 安装脚本
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-aarch64.sh

# 运行安装脚本
bash Miniconda3-latest-Linux-aarch64.sh

看到下图所示的内容反映 已经下载好了安装脚本 {F6BFA3ED-4696-467E-8D1A-BDAD6A28D87D}.png

输入运行脚本的命令后,一直按enter键,到最后输入yes即可完成安装 {B875EAFC-EAC4-4F1D-8B32-10EAC904EFD2}.png

出现下面这样的一个前缀符号说明配置了当前的环境,但是还没有使其生效,还需要继续按enter键才能使其继续安装

{50A8FCA6-0902-417A-BF2B-241561A70A1E}.png

出现这一个的时候,说明安装完毕了,接下来就需要安装pytorch了 {0426EBDD-C5E8-485D-A74B-DFF238C8FAB7}.png

通过输入下面的指令,来安装脚本生效

source ~/.bashrc

最后输入下面这行指令,当输出conda xx.xx.xx的版本号时,说明安装完成

conda --version

当看到前面有一个(base)说明安装和环境配置成功

处理后的中间图.png

  • 安装pytorch
  1. 配置虚拟环境
conda create -n cnndev python=3.10 -y

这一个命令的含义是,创建一个名字叫cnndev的虚拟环境,python的版本号为3.10 3. 激活虚拟环境

conda activate cnndev

通过这一行命令去激活这一个创造出来的环境,然后就会看到在账号名前面的(base)变成(cnndev) ,就说明激活成功了。

  1. 安装CPU版本

当前我的工作,暂时不需要用到NPU,因此只需要安装预编译的ARM64(aarch64)版本,后续会更新关于NPU的pytorch配置,但是其实NPU搭配的openblas的库也非常能打了。

# 升级 pip 到最新版
pip install --upgrade pip

# 安装 PyTorch 生态(会自动匹配 aarch64 版本的预编译 wheel)
pip install torch torchvision torchaudio

注意事项 : 当下载速度过慢时,需要考虑更换为国内的镜像源,输入下面的命令,可更换为清华的镜像源进行下载

# 使用清华大学的 pip 镜像源进行安装
pip install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple
  1. 测试安装是否成功
import torch

# 1. 检查版本
print(f"PyTorch Version: {torch.__version__}")

# 2. 检查当前环境是否能正常进行张量运算
x = torch.rand(5, 5)
y = torch.rand(5, 5)
z = torch.matmul(x, y)
print("矩阵乘法测试成功,结果形状:", z.shape)

# 3. 检查系统可用的 CPU 线程数(在鲲鹏这种多核服务器上很有用)
print(f"可用 CPU 线程数: {torch.get_num_threads()}")

输出的结果如下图所示

处理后的结果图.png

谢谢各位的点赞和评论~