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 系)也常遇到,速记备查。

生态矩阵(关键防坑表)

Installs
15
GitHub Stars
11
First Seen
Sep 27, 2026