How to specialize this skill (rewrite the body text at initialization time, without changing the name)
The body text below is thecurrently verified instance(MUSA / Ubuntu 22.04 x86_64 / S5000, 2026-09). When initializing for the first time on another platform, rewrite the body text according to the research conclusions from phase 3 of accel-init; after rewriting, this skill becomes that platform's version, and subsequent projects execute directly according to it. Machines that already have it installed do not chase newer versions in the repository (install-once policy; refreshing the repository is an explicit action).
- Rewrite section by section: keep the structure, delete entire sections that are meaningless for this platform; replace compatibility sets/package names/tool names with those of the actual platform.
- For verification commands and expected output, usethose verified on this machine; mark unverified ones as
(未验证); version lines should include the query date and the re-query command. - Platform names appear only in the content (tools, package names, indexes); the skill name and trigger words remain platform-independent.
- Facts such as local paths/usernames do not go into this skill; write them into the host machine layer.
Current instance: MUSA on Ubuntu 22.04 x86_64
Scope: portable. Use this for an arbitrary compatible Ubuntu 22.04 host;
current-server facts belong in the host-layer skill (local
<host>-environment / system-profile style factbook).
Use this skill for a new or existing Ubuntu 22.04 x86_64 host. Do not assume paths, package versions, GPU models, or driver state from another machine.
Operating principles
- Separate the kernel driver from the user-space SDK.
- Treat Toolkit, user-space
libmusa, muDNN, MCCL, torch, and torch_musa as one compatibility set. - Prefer a versioned user-space prefix so changing SDKs does not rewrite the host.
- Never upgrade or reload the kernel driver while GPU jobs are running.
- Validate dynamic linking before running a model.
- A successful
musaInfoquery does not prove that a torch_musa wheel is compatible.
Architecture
The kernel side normally contains:
mtgpukernel module.- GPU firmware.
- Device nodes and management integration.
- Exactly one active kernel driver version.
The selectable user-space side contains:
libmusa.so.1, analogous to the CUDA driver API library.- MUSA Toolkit and
libmusart.so. - muBLAS, muFFT, muSPARSE, muSOLVER, and related math libraries.
- muDNN.
- MCCL.
- Compiler and tools such as
mcc,musaInfo, andmusa_version_query. - Python, torch, and torch_musa.
Multiple user-space versions can coexist. The kernel module does not coexist in the same way.
Inspect before changing anything
Run read-only checks first:
lsb_release -a
uname -a
lscpu
lspci -nn
mthreads-gmi
lsmod | awk '$1 == "mtgpu" {print}'
dkms status
Record:
- Ubuntu point release and kernel.
- CPU architecture.
- GPU model and count.
- Kernel driver package/version.
- Existing
/usr/local/musa*,/opt/MUSA*, PATH, and LD_LIBRARY_PATH. - Active GPU processes and containers.
Do not infer a complete Toolkit install from mthreads-gmi alone.
Choose a compatibility set
Consult the current official release notes and torch_musa release page. Record a matrix with these columns before installing:
GPU model
compute capability
kernel driver
user-space libmusa
Toolkit
muDNN
MCCL target architecture
Python ABI
torch
torch_musa
Rules:
torchandtorch_musaversions must match exactly.- A wheel tagged
musa5.2.0expects the MUSA 5.2 user-space stack. - Use
mccl-s5000for S5000, not the S4000 build. - Python wheel tags such as
cp310andcp312must match the interpreter. - Do not let a normal PyPI torch wheel replace the MUSA torch wheel.
Official references:
https://docs.mthreads.com/musa-sdk/musa-sdk-doc-online/install_guidehttps://developer.mthreads.com/sdk/download/musahttps://github.com/MooreThreads/torch_musahttps://dl.mthreads.com/repo/api/pypi/pypi/simple/torch-musa/
Versioned user-space layout
Use a writable root chosen for that host:
export MUSA_ROOT="${MUSA_ROOT:-$HOME/opt/musa}"
export MUSA_VERSION=5.2.0
export MUSA_HOME="$MUSA_ROOT/$MUSA_VERSION"
export MUSA_DRIVER_LIB_DIR="$MUSA_ROOT/runtime-deps/$MUSA_VERSION/lib"
A useful layout is:
$MUSA_ROOT/
├── 4.3.5/
├── 5.1.0/
├── 5.2.0/
├── runtime-deps/
│ ├── 4.3.5/lib/
│ ├── 5.1.0/lib/
│ └── 5.2.0/lib/
└── downloads/
Do not assume the vendor installer supports this layout. Inspect install.sh -h first. If it accepts --prefix, use it. Otherwise extract DEBs or archives into a staging root and move only the user-space tree.
Download and integrity
Prefer the official offline SDK bundle or official Ubuntu repository.
For repository-based installation:
- Download the Moore Threads repository configuration DEB without installing it.
- Extract its source-list entry and keyring with
dpkg-deb -x. - Use an isolated APT state/cache directory under the chosen MUSA root.
- Verify
Release.gpgwith the extracted keyring. - Verify package hashes through APT or against the signed
Packagesindex.
For MUSA 5.2, the relevant package families include:
musa-toolkit-5-2
musa-musart-5-2
libmudnn3-musa-5-2
mccl-s5000
libmthreads-compute
Do not request mthreads-driver or mthreads-dkms for a user-space-only install.
Download dependency closure with APT in download-only mode, then extract each DEB:
dpkg-deb -x package.deb "$STAGING_ROOT"
Packages commonly place the Toolkit under:
$STAGING_ROOT/usr/local/musa-X.Y
Move that directory to the selected $MUSA_HOME. Keep libmusa.so.1 from libmthreads-compute in the matching versioned runtime directory.
Manual environment configuration
Environment variables are the portable interface. Scripts and modules are optional wrappers around these settings:
export MUSA_HOME="$MUSA_ROOT/$MUSA_VERSION"
export MUSA_INSTALL_PATH="$MUSA_HOME"
export PATH="$MUSA_HOME/bin${PATH:+:$PATH}"
export LD_LIBRARY_PATH="$MUSA_DRIVER_LIB_DIR:$MUSA_HOME/lib${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}"
Ordering matters:
- Matching user-space
libmusa.so.1directory. - Matching Toolkit/muDNN/MCCL library directory.
- Existing inherited library paths.
Set visible GPUs separately:
export MUSA_VISIBLE_DEVICES="${MUSA_VISIBLE_DEVICES:?set the allocated device list}"
CUDA_VISIBLE_DEVICES is not a portable substitute for native MUSA code.
For shared HPC systems, Environment Modules can set the same variables. For project-local workflows, an explicit envs.sh can set them. Do not hide critical MUSA variables only inside Conda activation hooks.
MPI, MKL, and Python dependencies
Some MUSA torch wheels directly require:
libmpi.so.40
libmkl_intel_lp64.so.2
libmkl_gnu_thread.so.2
libmkl_core.so.2
Check the wheel rather than assuming:
ldd /path/to/torch/lib/libtorch_global_deps.so
ldd /path/to/torch_musa/lib/libmusa_python.so.*
OpenMPI and MKL may be installed in the Conda environment, a versioned local runtime directory, or the system. Keep their library directories explicit in LD_LIBRARY_PATH when they are not registered with the system loader.
If torch reports that an extension compiled against NumPy 1.x cannot run with NumPy 2.x, use a compatible numpy<2 version or rebuild the extension. Python 3.12 commonly uses numpy==1.26.4 for this compatibility case.
Install Python packages
Create a clean Python environment with a supported ABI. Install the matching MUSA torch and torch_musa wheels from the official Moore Threads index. Keep application dependencies separate from the accelerator stack.
After installation, record:
python -V
python -m pip show torch torch_musa
python -m pip check
Do not install unmodified CUDA dependency files containing nvidia-*, CUDA Triton, bitsandbytes, BitBLAS, NCCL, or CUDA-only custom extensions into a MUSA environment.
Validation ladder
Validate from lowest to highest risk.
Toolkit and driver API:
which mcc musaInfo musa_version_query mccl_version
musa_version_query
musaInfo
mccl_version
mcc --version
Dynamic libraries:
ldd /path/to/libmusa_python.so | awk '/not found/ {print}'
ldd /path/to/libtorch_global_deps.so | awk '/not found/ {print}'
Single-device torch:
python - <<'PY'
import torch
import torch_musa
print(torch.__version__)
print(torch_musa.__version__)
print(torch.musa.is_available())
print(torch.musa.device_count())
x = torch.randn(64, 64, device="musa", dtype=torch.bfloat16)
y = x @ x
torch.musa.synchronize()
print(y.shape, y.dtype)
PY
Then validate, in order:
- One-process MCCL.
- Two-process
torchrunMCCL all-reduce. - Short application workload.
- All-device communication.
- Long-running training stability.
Failure patterns
libmpi.so.40 missing:
- Install/extract an OpenMPI 4.1 runtime or the version required by the wheel.
- Check secondary dependencies of
libmpi.so.40withldd.
MKL libraries missing:
- Install/extract the MKL runtime libraries required by
libtorch_global_deps.so. - Do not solve only the first missing library and stop checking.
- If torch import then fails on
iJIT_NotifyEvent, the selected MKL is ABI-incompatible with the wheel. Pin a compatible older MKL row and record it; do not assume the newest MKL with matching SONAMEs is compatible.
undefined symbol from muBLAS, muDNN, or torch_musa:
- A mismatched
libmusa, Toolkit, or wheel is being loaded. - Inspect
LD_LIBRARY_PATHorder andlddoutput. - Do not copy a random
.sointo the Python package.
musaInfo works but torch import fails:
- Toolkit queries prove only the C/C++ user stack.
- Verify torch, torch_musa, Python ABI, MPI, MKL, and NumPy separately.
No devices or kernel errors:
- Stop user-space debugging and inspect kernel driver, firmware, IOMMU/VT-d requirements, device nodes, and hardware support.
- Schedule driver changes only after draining GPU jobs.
Completion criteria
An environment is complete only when:
- Versions and installation source are documented.
PATHandLD_LIBRARY_PATHare explicit and reproducible.lddreports no missing libraries.- Single-device compute succeeds.
- Required MCCL topology succeeds.
- The real application completes a short smoke workload.
- Rollback consists of selecting another prefix or deleting the local directory, not repairing the host package database.