在PAI中使用PPU PIP源（v1.7.x）-PG1阿里云产品-阿里云

1. PPU PIP源简介

PPU PIP源提供为SAIL产品优化的框架、基于PPU SDK重编译的开源库，支持用户通过pip install的方式安装软件包，方便用户快速搭建生产环境。如果客户需要的包不在PPU PIP源内，则默认会从官方PYPI源拉取软件包。
PPU PIP源上的软件包会随着正式PPU SDK release发布，也会按照周更的节奏持续更新。
PPU-PIP包清单会持续更新基于PPU SDK v1.7.x 的PIP源迭代。

PPU-PIP包清单为适用于PPU SDK v1.7.x release的全部pip安装包一览表，其中“ce”基于SAIL SDK编译的开源框架安装包，“oe”表示针对SAIL产品做过性能优化的框架安装包。

https://art-pub.eng.t-head.cn/artifactory/api/pypi/ptgai-pypi_ppu_	ubuntu_	cu118_	index/simple/
源固定地址	目前只区分ubuntu和alios，不区分os具体版本	表示cuda11.8	固定格式

新地址中不再包含OS/CUDA信息，从SAIL PIP 1.7.x开始，PIP软件包全部上传到同一个地址，安装过程中PIP包会自动识别生产环境，安装与客户环境下CUDA、Torch、Python相匹配的包。

https://art-pub.eng.t-head.cn/artifactory/api/pypi/pypi_index/simple

能看到，对比老的 OS/CUDA 绑定模式，新的源地址上更加简洁。

在未来一段时间内，使用老的 OS/CUDA 绑定模式地址依然可以拉到新发布的包，用户可以根据自己的需求选择使用。

以 pip install apex==25.4+ppu1.7.0.ce 为例：

25.4

+ppu

1.7.0

.ce 或者 .oe

框架版本号

固定格式

发布版本

ce：开源（Community Edition）

oe：针对PPU优化（Optimized Edition）

ce、oe缺省：不指定包的类型，直接使用指令 pip install apex==25.4+ppu1.7.0 安装也可以安装成功。如果当前配置下即存在开源的PIP包又存在针对SAIL产品优化过的PIP包，默认安装的是优化过的包。
发布版本缺省：不指定发布版本，直接使用指令 pip install apex==25.4 安装，默认安装SAIL PIP源上最近一次发布的apex25.4安装包。
版本号缺省：如果不指定版本号，直接使用 pip install apex 安装，会默认安装一个SAIL PIP源上框架版本最新的apex，即apex25.9。

应用场景：在SAIL SDK 1.7.0 发布的docker：llm:pytorch2.7.0-ubuntu24.04-cuda12.8-sglang0.5.2-py312 中安装apex 25.4。

pip install apex==25.9+ppu1.7.0 -i https://art-pub.eng.t-head.cn/artifactory/api/pypi/pypi_index/simple

注意，这里需要手动加上+ppu1.7.0！

SAIL SDK本身具备一定的兼容性，所以 pip源上的whl包，尤其是ce开源编译软件包，在1.7.0 SDK Release中发布一次即可，无需在如1.7.1这样的小版本迭代中再次发布。

pip install apex==25.9 -i https://art-pub.eng.t-head.cn/artifactory/api/pypi/pypi_index/simple

如遇到以下pip源hash值错误，请使用--no-cache指令解决。

pip install --no-cache-dir torch==2.1.2+ppu1.4.0

也可以使用pip cache purge指令清除当前系统中全部pip缓存。

pip cache purge