MiroPsota/torch_packages_builder
Builder and index for PyTorch packages
解决的问题
简化包含自定义算子的 PyTorch 基础包的构建与分发流程。这些包通常需要特定的硬件架构和 PyTorch 版本才能运行，对普通用户而言编译和托管非常困难。
工作原理
该项目提供两个自动化的 GitHub 工作流：一个是在多种架构上编译带有自定义算子的 PyTorch 包，并将它们作为 wheel 发布到 GitHub 发布页面；另一个是创建一个符合 PEP 503 标准的包索引，托管在 GitHub Pages 上，允许用户通过 pip install --extra-index-url 安装这些预构建的二进制文件。
适用人群
需要安装带有自定义算子的 PyTorch 扩展，但不想在本地从源码编译的开发者和研究人员。
主要亮点
- 自动构建带有自定义算子的 PyTorch 包。
- 符合 PEP 503 标准的包索引，便于与 pip 集成。
- 支持多种计算平台，包括 CPU、CUDA 和 ROCm。
- 使用构建认证（build attestation）为 wheel 建立可追溯性。
相关
- 项目
- 项目
- 项目
- 项目
- Dispatch