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