MiroPsota/torch_packages_builder
Builder and index for PyTorch packages
何を解決するか
カスタム演算子を含むPyTorchベースのパッケージのビルドと配布プロセスを簡素化します。これらのパッケージは特定のハードウェアアーキテクチャやPyTorchバージョンを必要とするため、一般ユーザーにとってコンパイルやホスティングが困難です。
動作方法
このプロジェクトは、2つの自動化されたGitHubワークフローを提供します。1つは、さまざまなアーキテクチャでカスタム演算子付きのPyTorchパッケージをコンパイルし、GitHubリリースにwheelとして公開するもの。もう1つは、GitHub Pages上でホストされるPEP 503準拠のパッケージインデックスを作成し、ユーザーが pip install --extra-index-url でこれらの事前ビルド済みバイナリをインストールできるようにするものです。
対象ユーザー
ローカルでソースからコンパイルせずに、カスタム演算子付きのPyTorch拡張をインストールしたい開発者や研究者。
主な特徴
- カスタム演算子付きのPyTorchパッケージの自動ビルド。
- pip統合を容易にするPEP 503準拠のパッケージインデックス。
- CPU、CUDA、ROCmを含む複数のコンピューティングプラットフォームのサポート。
- wheelの信頼性を確立するためのビルド認証の利用。
関連
- プロジェクト
- プロジェクト
- プロジェクト
- プロジェクト
- Dispatch