kb-accel-sdk
Installation
SKILL.md
加速平台 SDK 与生态知识库
知识类 skill:内容为跨平台公共知识(调研日期 2026-09-27;Ascend 节 2026-09-28 经 910B2 实机初始化验证刷新,易过期条目附重查入口)。本机特有事实不写这里(宿主机层管)。结论与实机冲突时以实机为准,并回流更新本库。
平台总览
| 平台 | 厂商 | 软件栈 | smi | torch 后端 | 通信库 | wheel 来源 |
|---|---|---|---|---|---|---|
| CUDA | NVIDIA | 驱动 + CUDA Toolkit(nvcc/cuBLAS/cuDNN/NCCL/TensorRT) | nvidia-smi | 原生 | NCCL | PyPI / download.pytorch.org/whl/cuXXX |
| MUSA | 摩尔线程 | mtgpu 驱动 + MUSA SDK(musart/toolkit/muDNN/MCCL) | mthreads-gmi | torch_musa | MCCL | dl.mthreads.com 厂商 index |
| Ascend | 华为 | 驱动+固件 + CANN(toolkit/kernels/NNAL,含 HCCL、ATB;推理另有 MindIE) | npu-smi | torch_npu | HCCL | PyPI(torch_npu) + 华为云 mirrors.huaweicloud.com/ascend(triton-ascend/离线包) |
| ROCm | AMD | amdgpu 驱动 + ROCm(HIP/hipBLAS/MIOpen/RCCL) | rocm-smi | 原生(HIP 转译,device 仍叫 cuda) | RCCL | download.pytorch.org/whl/rocmX.Y |
| PPU | 阿里平头哥 | 真武 810/810E,GPGPU 路线;公开 SDK 文档极少(2026-01 才官方证实存在) | (未验证) | (未验证) | (未验证) | 阿里云渠道,无公共源 |
| 沐曦 MACA | 沐曦 | MXMACA(mcBLAS/mcDNN/mcFlashAttention) | mx-smi | mcPytorch | mcCL(未验证) | 沐曦开发者社区镜像/发布包 |
寒武纪 MLU(Neuware/torch_mlu)、海光 DCU(DTK,ROCm 系)也常遇到,速记备查。