TileLang 怎么下载?DeepSeek 开源的昇腾 NPU 编程语言 wheel 包、PyPI 与源码获取方式
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 源码仓库,四条装出来的是同一个包,差别只在版本新旧与是否需要本地编译(来源)。 按你的场景选:
- 要最快跑起来 → 走 PyPI。执行
pip install tilelang,pip 会自动挑一个与当前 Python 匹配的 wheel(来源)。预期结果:装完终端没有报错。 - 要离线分发或固定版本 → 走 GitHub Release。打开
tile-ai/tilelang的 Release 页,下载对应平台的 wheel 文件,本地执行pip install tilelang-0.0.0.dev0+ubuntu.20.4.cu120-py3-none-any.whl(文件名以你下到的为准)。预期结果:因为不访问网络,等价于一次离线安装(来源)。 - 要跟主分支代码 → 走源码仓库。执行
pip install git+https://github.com/tile-ai/tilelang.git,让 pip 直接拉仓库构建。预期结果:装到的版本可能比 PyPI 上的正式版更新,但构建耗时明显更长(来源)。 - 想抢先试新特性 → 走 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(来源)。 按下面顺序做:
- 先备好昇腾环境。确认机器上已有 CANN(至少 8.3.RC1)与 torch-npu(至少 2.6.0.RC1),然后
source {your-cann-installed-path}/ascend-toolkit/set_env.sh载入环境变量。预期结果:echo $ASCEND_HOME_PATH能打印出昇腾工具链路径。 - 能联网就直接装。执行
pip install tilelang-ascend。预期结果:pip 从 PyPI 拉包并装完,不需要本地编译。 - 要离线就用官方 wheel。把下到的 wheel 放到目标机器上,设置
export ASCEND_HOME_PATH=/usr/local/Ascend/ascend-toolkit/latest,再执行pip install tilelang-*.whl。预期结果:wheel 文件名通配到的那一个包被装上,路径按你机器的实际安装位置改。 - 要自己构建就用官方脚本。
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 标签对不对(来源)。 具体做法:
- 确认 import 与版本。执行
python -c "import tilelang; print(tilelang.__version__)"。预期结果:打印出版本号;报ModuleNotFoundError说明包装在别的 Python 环境里,先用which python和pip -V对齐。 - 看 wheel 的本地版本标签。官方构建时会把 SDK 与工具链信息嵌进版本号,形如
tilelang-0.1.6.post1+cu116.git0d4a74be-cp38-abi3-linux_x86_64.whl,其中+cu116表示 CUDA 11.6 构建,+rocm与+metal同理(来源)。预期结果:标签与你的驱动 / SDK 对得上;对不上就换一个 wheel。 - 跑一个真实算子。昇腾侧进
examples/gemm执行python example_gemm.py。预期结果:终端打印Kernel Output Match!,说明下载的包能编出 kernel 并跑通(来源)。
下载 TileLang 的注意事项
- 认准
tile-ai与 PyPI 官方页。TileLang 的仓库与包都在tile-ai名下;第三方聚合站或「汉化包」不是官方分发点,本文只覆盖官方渠道。 - 分清 TileLang 与 DeepSeek 的关系。TileLang 由北京大学团队发起,DeepSeek 于 2026-09-30 开源的是其面向昇腾算力平台的基础设施组件,其中包含 TileLang 昇腾版,并说明 DeepSeek V4 系列模型训练多数算子基于该工具实现——它不是 DeepSeek 自研语言,写文档、建索引时别搞错归属。
- 下载完先别急着升版本。刚下完就换版本容易把「环境问题」和「版本问题」搅在一起,先把当前版本跑通,再按《TileLang 怎么更新》升级。
包拿到手之后,下一步是把它装进环境并验证两条硬件路线,走《TileLang 怎么安装》;常用命令与工具链速查在《TileLang 命令大全》。
来源:TileLang Installation Guide、tilelang · PyPI、tilelang-ascend · PyPI、tile-ai/tilelang Releases
常见问题
TileLang 有四个官方取包渠道:PyPI、GitHub Release 预编译 wheel、nightly 预览索引和源码仓库。想最快用上就执行 pip install tilelang;要离线分发或固定版本就下载 Release 里的 wheel 再本地安装;要跟主线代码就装 git+https://github.com/tile-ai/tilelang.git。四条渠道装出来的是同一个包,差别只在版本新旧和是否需要本地编译。
昇腾版 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 文件名会把 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 装完用一行命令就能确认:python -c "import tilelang; print(tilelang.__version__)",能打印出版本号说明当前解释器已经能找到它。这一步报 ModuleNotFoundError,通常是装进了另一个 Python 环境,先用 which python 与 pip -V 对齐解释器再重装。想验证真的能跑,进 examples/gemm 执行 python example_gemm.py,成功会打印 Kernel Output Match!。
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
来源
- TileLang Installation Guide· TileLang (tile-ai)
- tilelang · PyPI· PyPI
- tilelang-ascend · PyPI· PyPI
- tile-ai/tilelang Releases· GitHub