TileLang 怎么下载?DeepSeek 开源的昇腾 NPU 编程语言 wheel 包、PyPI 与源码获取方式

安装与快速上手发布于 2026-10-01作者: DeepSeek Plugin 插件市场
TileLang昇腾Ascend NPUtilelang 下载pipwheelDeepSeek
TileLang 是北大团队发起、DeepSeek 于 2026-09-30 开源昇腾适配版的 GPU kernel 编程语言。本文讲清四条官方下载渠道——PyPI、GitHub Release wheel、nightly 索引与源码仓库,以及昇腾版 tilelang-ascend 的获取方式。

TileLang 的官方取包渠道有四条:PyPI 上 pip install tilelang、GitHub Release 里的预编译 wheel、nightly 预览索引,以及源码仓库(可直接 pip 装 git 地址)。昇腾 NPU 走另一条线——tilelang-ascend,可以直接 pip 装,也可以下 wheel 或用 build_wheel_ascend.sh 自己构建。

本文按「渠道 → 昇腾版怎么拿 → 下完怎么核对」三段讲,只解决「包从哪来」。装进环境要备什么、装哪条路线,见《TileLang 怎么安装》。

TileLang 从哪下载?四条官方渠道与各自适用场景

TileLang 由官方维护四条下载渠道,都在 tile-ai 名下:PyPI 包、GitHub Release 的预编译 wheel、nightly 预览索引、GitHub 源码仓库,四条装出来的是同一个包,差别只在版本新旧与是否需要本地编译(来源)。 按你的场景选:

  1. 要最快跑起来 → 走 PyPI。执行 pip install tilelang,pip 会自动挑一个与当前 Python 匹配的 wheel(来源)。预期结果:装完终端没有报错。
  2. 要离线分发或固定版本 → 走 GitHub Release。打开 tile-ai/tilelang 的 Release 页,下载对应平台的 wheel 文件,本地执行 pip install tilelang-0.0.0.dev0+ubuntu.20.4.cu120-py3-none-any.whl(文件名以你下到的为准)。预期结果:因为不访问网络,等价于一次离线安装(来源)。
  3. 要跟主分支代码 → 走源码仓库。执行 pip install git+https://github.com/tile-ai/tilelang.git,让 pip 直接拉仓库构建。预期结果:装到的版本可能比 PyPI 上的正式版更新,但构建耗时明显更长(来源)。
  4. 想抢先试新特性 → 走 nightly 索引。执行 pip install tilelang -f https://tile-ai.github.io/whl/nightly。预期结果:装上官方当天构建的预览版,稳定性低于正式发布版,本文不推荐在生产环境用。

想自己编译而非装现成包,官方还提供了容器路径:git clone --recursive https://github.com/tile-ai/tilelang 后进 docker/ 目录,docker build -f Dockerfile.cu120 -t tilelang-cu120 . 构建镜像(来源)。

昇腾 NPU 版 TileLang 怎么下载?tilelang-ascend 的三种拿法

昇腾平台上的 TileLang 以独立包 tilelang-ascend 分发,官方给了三条获取路径:PyPI 安装、下载预编译 wheel、用 build_wheel_ascend.sh 从源码自建 wheel(来源)。 按下面顺序做:

  1. 先备好昇腾环境。确认机器上已有 CANN(至少 8.3.RC1)与 torch-npu(至少 2.6.0.RC1),然后 source {your-cann-installed-path}/ascend-toolkit/set_env.sh 载入环境变量。预期结果:echo $ASCEND_HOME_PATH 能打印出昇腾工具链路径。
  2. 能联网就直接装。执行 pip install tilelang-ascend。预期结果:pip 从 PyPI 拉包并装完,不需要本地编译。
  3. 要离线就用官方 wheel。把下到的 wheel 放到目标机器上,设置 export ASCEND_HOME_PATH=/usr/local/Ascend/ascend-toolkit/latest,再执行 pip install tilelang-*.whl。预期结果:wheel 文件名通配到的那一个包被装上,路径按你机器的实际安装位置改。
  4. 要自己构建就用官方脚本。git clone --recursive https://github.com/tile-ai/tilelang-ascend.git 后进目录,执行 ./build_wheel_ascend.sh [--enable-llvm],产物在 dist/ 下,再用 pip install dist/tilelang-*.whl 装上。预期结果:dist/ 里出现新的 wheel,安装成功(来源)。

官方说明 tilelang-ascend 目前针对 A2 与 A3 两类 NPU 做过验证,其它昇腾型号属于「尽力兼容」(来源)。

下载后怎么核对?TileLang 版本、wheel 标签与 import 验证

核对三件事就够了:解释器能不能 import、版本号是多少、wheel 文件名里的 SDK 标签对不对(来源)。 具体做法:

  1. 确认 import 与版本。执行 python -c "import tilelang; print(tilelang.__version__)"。预期结果:打印出版本号;报 ModuleNotFoundError 说明包装在别的 Python 环境里,先用 which python 和 pip -V 对齐。
  2. 看 wheel 的本地版本标签。官方构建时会把 SDK 与工具链信息嵌进版本号,形如 tilelang-0.1.6.post1+cu116.git0d4a74be-cp38-abi3-linux_x86_64.whl,其中 +cu116 表示 CUDA 11.6 构建,+rocm 与 +metal 同理(来源)。预期结果:标签与你的驱动 / SDK 对得上;对不上就换一个 wheel。
  3. 跑一个真实算子。昇腾侧进 examples/gemm 执行 python example_gemm.py。预期结果:终端打印 Kernel Output Match!,说明下载的包能编出 kernel 并跑通(来源)。

下载 TileLang 的注意事项

  1. 认准 tile-ai 与 PyPI 官方页。TileLang 的仓库与包都在 tile-ai 名下;第三方聚合站或「汉化包」不是官方分发点,本文只覆盖官方渠道。
  2. 分清 TileLang 与 DeepSeek 的关系。TileLang 由北京大学团队发起,DeepSeek 于 2026-09-30 开源的是其面向昇腾算力平台的基础设施组件,其中包含 TileLang 昇腾版,并说明 DeepSeek V4 系列模型训练多数算子基于该工具实现——它不是 DeepSeek 自研语言,写文档、建索引时别搞错归属。
  3. 下载完先别急着升版本。刚下完就换版本容易把「环境问题」和「版本问题」搅在一起,先把当前版本跑通,再按《TileLang 怎么更新》升级。

包拿到手之后,下一步是把它装进环境并验证两条硬件路线,走《TileLang 怎么安装》;常用命令与工具链速查在《TileLang 命令大全》。

来源:TileLang Installation Guide、tilelang · PyPI、tilelang-ascend · PyPI、tile-ai/tilelang Releases

常见问题

TileLang 官方下载渠道有几个?PyPI、GitHub Release 与源码包该选哪个?

TileLang 有四个官方取包渠道:PyPI、GitHub Release 预编译 wheel、nightly 预览索引和源码仓库。想最快用上就执行 pip install tilelang;要离线分发或固定版本就下载 Release 里的 wheel 再本地安装;要跟主线代码就装 git+https://github.com/tile-ai/tilelang.git。四条渠道装出来的是同一个包,差别只在版本新旧和是否需要本地编译。

昇腾 NPU 上的 TileLang 怎么下载?tilelang-ascend 有预编译 wheel 吗?

昇腾版 TileLang 走 tilelang-ascend 这个独立包,可以直接 pip install tilelang-ascend,也可以下载官方 wheel 后用 pip install tilelang-*.whl 本地安装。官方还提供 build_wheel_ascend.sh,让你在内网自己构建 wheel。构建前先确认 CANN 不低于 8.3.RC1、torch-npu 不低于 2.6.0.RC1,并 source 昇腾工具链的 set_env.sh。

下载 TileLang 的 wheel 时,怎么判断这个包跟我的 CUDA 版本对得上?

TileLang 的 wheel 文件名会把 SDK 与工具链版本写进本地版本标签,例如 tilelang-0.1.6.post1+cu116.git0d4a74be-cp38-abi3-linux_x86_64.whl 里的 +cu116 表示 CUDA 11.6 构建。认准 +cu…、+rocm、+metal 这类标签,再核对 Python 版本与 glibc 2.28 的要求即可。官方也说明可以用 NO_VERSION_LABEL 关掉这个标签,所以文件名里没有标签不代表不是官方产物。

TileLang 下载完之后怎么确认装好了、装的是哪个版本?

TileLang 装完用一行命令就能确认:python -c "import tilelang; print(tilelang.__version__)",能打印出版本号说明当前解释器已经能找到它。这一步报 ModuleNotFoundError,通常是装进了另一个 Python 环境,先用 which python 与 pip -V 对齐解释器再重装。想验证真的能跑,进 examples/gemm 执行 python example_gemm.py,成功会打印 Kernel Output Match!。

内网机器拿不到 PyPI,TileLang 有什么离线下载的办法?

TileLang 支持离线获取:在外网机器上从 GitHub Release 页下载 wheel 文件,拷进内网后用 pip install tilelang-xxx.whl 安装,全程不需要访问 PyPI。昇腾侧同理,用 wheel 包或先在内网外构建好 wheel 再拷入。官方 Installation 文档也给了源码构建路径,源码包补齐 TVM 子模块后可离线编译。

相关术语

TileLang
TileLang(tile-lang)是由北京大学团队发起、基于 TVM 编译栈的 tile 级领域特定语言,用类 Python 语法生成高性能 GPU / NPU kernel。DeepSeek 于 2026-09-30 宣布开源其昇腾适配版,并说明 DeepSeek V4 系列模型训练多数算子基于该工具实现。— TileLang Installation Guide
tilelang-ascend
tilelang-ascend 是 TileLang 面向华为昇腾 NPU 的专用发行版,编译器后端支持 Ascend C & PTO 与 AscendNPU IR 两条技术路线。它要求 CANN 8.3.RC1 及以上、torch-npu 2.6.0.RC1 及以上。— tilelang-ascend · PyPI
nightly wheel
nightly wheel 是 TileLang 官方从主分支持续构建的预览版安装包,通过 pip install tilelang -f https://tile-ai.github.io/whl/nightly 安装。它包含最新代码改动,稳定性低于正式发布版。— TileLang Installation Guide
ASCEND_HOME_PATH
ASCEND_HOME_PATH 是昇腾工具链的安装根路径环境变量,安装 tilelang-ascend 前通常指向 /usr/local/Ascend/ascend-toolkit/latest。它和 CANN 的 set_env.sh 一起决定编译器能否找到昇腾底层库。— tilelang-ascend · PyPI

来源