跳到主要内容

安装 S5000 5.2.0

本文档介绍如何在服务器系统上安装 Linux Server Driver,支持在线安装(APT)和离线安装两种方式。按照本文档完成操作后,您可以完成驱动安装并验证驱动是否正常生效。

概览

平台支持

MTT S5000

CPU操作系统版本包类型
INTEL(R) XEON(R) GOLD 6530
Intel(R) Xeon(R) Gold 6430
AlinuxAlibaba Cloud Linux 3RPM
INTEL(R) XEON(R) GOLD 6530VesselOSVesselOS 2.0 (LTS-SP2)RPM
Hygon C86-4GKylin Linux Advanced ServerV10/V11RPM
Hygon C86-4GBigCloud Enterprise Linux For Euler21.10 LTSRPM
INTEL(R) XEON(R) GOLD 6530
Hygon C86-4G
AMD EPYC 9T34 64-Core
Ubuntu Server22.04DEB
INTEL(R) XEON(R) GOLD 6530Ubuntu Server24.04.1 LTSDEB

安装包说明

根据系统架构和包管理系统选择合适的安装包:

架构系统类型安装包
x86_64Ubuntu 22.04 / 24.04musa_5.2.0-server_amd64.deb
x86_64RPM 系musa_5.2.0-server_amd64.rpm

安装流程

建议按以下顺序完成安装。下表列出了主要步骤及其说明:

步骤操作说明是否必需
1安装前准备安装依赖并确认系统已识别显卡
2安装驱动根据平台和安装方式完成驱动安装
3安装后配置配置用户组权限并验证驱动是否正常生效建议
4不重启更换驱动适用于特殊场景下的驱动更换流程按需
5安装 MUSA Toolkit如需继续使用开发工具链,可继续安装 Toolkit按需

安装前准备

1. 依赖安装

对于 Ubuntu 系统,本节依赖安装主要适用于离线安装方式。如果您选择在 Ubuntu 22.04 系统中使用在线安装(APT)方式,可跳过本节,直接前往 Ubuntu Server 中的“在线安装(APT)”章节。

Ubuntu 系统

# 安装依赖
sudo apt install -y clang libclang-dev libelf-dev gcc-multilib freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev libglew-dev libglfw3 libglfw3-dev libfreeimage3 libfreeimage-dev libtiff-dev libboost-all-dev ninja-build cmake ocl-icd-opencl-dev libglew-dev libnuma-dev
sudo apt install -y python3-pip libstdc++-12-dev
sudo apt -y install python-is-python3
sudo apt install -y python3-pip
sudo pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple allure-pytest pandas paramiko ahocorapy

# 安装 dkms
sudo apt install -y dkms

RPM 系统(Alinux / VesselOS / Kylin / Bclinux)

# 查看 PCI 是否识别到显卡
sudo lspci -nn | grep -E " 1ed5 | Moore "

# 安装 dkms(若已安装请忽略)
sudo dnf -y install dkms
# 或
sudo yum -y install dkms

2. 验证显卡识别

安装驱动前,确认系统已识别到显卡:

sudo lspci -nn | grep -E " 1ed5 | Moore "

安装驱动

Ubuntu Server

此方式适用于 Ubuntu 22.04 APT 环境。若不希望手动下载离线安装包,可通过 Mthreads APT 软件源在线安装 mthreads-driver。Ubuntu 24.04 和 RPM 系统请使用离线安装方式。

步骤 1:配置 APT 软件源

wget https://dl.mthreads.com/repo/repository/ubuntu2204/pool/jammy/amd64/musa-repo-jammy_1.0.0.11-1_all.deb
sudo dpkg -i musa-repo-jammy_1.0.0.11-1_all.deb

步骤 2:更新本地缓存

sudo apt update

步骤 3:安装驱动

完成 APT 软件源配置并更新本地缓存后,可直接在线安装 Linux Driver。

sudo apt install mthreads-driver

步骤 4:重启系统

sudo reboot
提示

卸载 mtgpu 模块时,如提示 mtgpu is in use,请先执行 lsmod | grep mtgpu 查看依赖关系,先卸载依赖 mtgpu 的其他模块,再重新执行卸载命令;如仍无法卸载,请重启系统后重试。

RPM 系统(Alinux / VesselOS / Kylin / Bclinux)

步骤 1:卸载旧版本(如已安装)

# 卸载旧 musa rpm 包
sudo rpm -e musa

# 卸载 mtgpu 模块
sudo rmmod mtgpu

# 查询是否仍有 mtgpu 模块残留
lsmod | grep mtgpu

步骤 2:安装驱动

# x86_64 架构
sudo rpm -ivh musa_5.2.0-server_amd64.rpm

步骤 3:重启系统

sudo reboot

安装后配置

1. 设置用户组权限

非 Root 用户如需获取设备使用权限,需要将用户添加到 render 和 video 组:

# 把普通用户账号加入 render 和 video 组
sudo usermod -aG render ${USER}
sudo usermod -aG video ${USER}

修改权限后需要重新登录账户使更改生效。

2. 验证安装

方式一:使用 mthreads-gmi 工具

mthreads-gmi

该命令将显示 GPU 状态信息,包括设备型号、驱动版本、显存使用等。

方式二:检查驱动模块

# 检查驱动模块是否加载
lsmod | grep mtgpu

预期输出应包含 mtgpu 模块。

方式三:查看驱动版本

# 离线 DEB 安装方式
sudo dpkg -s musa

# 在线 APT 安装方式
sudo dpkg -s mthreads-driver

# RPM 系统
sudo rpm -qi musa

方式四:检查显卡识别

sudo lspci -nn | grep -E " 1ed5 | Moore "

应能看到摩尔线程显卡设备。

不重启更换驱动

特殊情况下如需在不重启系统的情况下更换驱动:

步骤 1:卸载旧驱动

# Ubuntu 在线安装(APT)方式
sudo apt remove mthreads-driver

# DEB 系统
sudo dpkg -P musa

# RPM 系统
sudo rpm -e musa

步骤 2:卸载驱动模块

# 卸载 mtgpu 模块
sudo modprobe -rv mtgpu

# 验证是否成功卸载
lsmod | grep mtgpu

如提示 mtgpu is in use,请先执行 lsmod | grep mtgpu 查看依赖关系,先卸载依赖 mtgpu 的其他模块,再重新执行该命令。

步骤 3:安装新驱动

# Ubuntu 在线安装(APT)方式
sudo apt install mthreads-driver

# Ubuntu 22.04 / 24.04
sudo dpkg -i musa_5.2.0-server_amd64.deb

# RPM 系统
sudo rpm -ivh musa_5.2.0-server_amd64.rpm

步骤 4:重新加载驱动模块

# 重新加载 kmd
sudo modprobe -v mtgpu

# 验证是否成功加载
lsmod | grep mtgpu

常见问题

Q1: 安装失败 - 依赖问题
# Ubuntu 22.04 / 24.04
sudo apt --fix-broken install
sudo dpkg -i musa_5.2.0-server_amd64.deb

# RPM 系统
sudo dnf install -y <缺失的依赖包>
sudo rpm -ivh musa_5.2.0-server_amd64.rpm
Q2: 模块加载失败
  • 检查内核版本兼容性
  • 查看内核日志:dmesg | grep mtgpu
  • 尝试手动加载:sudo modprobe -v mtgpu
  • 检查 dkms 是否成功编译:dkms status 通过 cat /proc/cmdline 查询 intel_iommu=on iommu=pt
Q3: 显卡未识别
  • 检查显卡是否正确插入 PCIe 插槽
  • 检查电源连接是否牢固
  • 尝试重新插拔显卡
  • 检查 BIOS 中 PCIe 设置
Q4: 用户权限不足
# 确保用户已加入正确的组
sudo usermod -aG render ${USER}
sudo usermod -aG video ${USER}

# 重新登录使更改生效

MUSA Toolkit 安装

驱动安装完成后,如需使用 MUSA Toolkit、muDNN、mccl 等组件,请参考相关安装指南。

1. 前提条件

  • 已安装兼容版本 MT GPU 驱动

  • 已验证驱动正常工作

2. 安装 MUSA Toolkit

此方式当前适用于 Ubuntu 22.04 APT 环境。Ubuntu 24.04 和 RPM 系统暂不支持在线安装,请使用离线安装方式。

若已完成前文 APT 软件源配置,可直接在线安装 MUSA Toolkit。

sudo apt install musa-toolkit

3. 验证 Toolkit 安装

# 查询版本
musa_version_query

# 查看 GPU 信息
musaInfo