MiroPsota/torch_packages_builder
Builder and index for PyTorch packages
해결하는 문제
커스텀 연산자를 포함한 PyTorch 기반 패키지의 빌드 및 배포 과정을 단순화합니다. 이러한 패키지는 특정 하드웨어 아키텍처와 PyTorch 버전이 필요하므로 일반 사용자에게 컴파일 및 호스팅이 어려운 경우가 많습니다.
작동 방식
이 프로젝트는 두 가지 자동화된 GitHub 워크플로우를 제공합니다. 하나는 다양한 아키텍처에서 커스텀 연산자와 함께 PyTorch 패키지를 컴파일하고 GitHub 릴리스에 wheel로 게시하는 작업이며, 다른 하나는 GitHub Pages에 호스팅되는 PEP 503 준수 패키지 인덱스를 생성하여 pip install --extra-index-url을 통해 사용자가 사전 빌드된 바이너리를 쉽게 설치할 수 있도록 하는 것입니다.
대상 사용자
로컬에서 소스에서 컴파일하지 않고도 커스텀 연산자와 함께 PyTorch 확장 기능을 설치하고자 하는 개발자 및 연구자입니다.
주요 특징
- 커스텀 연산자와 함께 PyTorch 패키지의 자동 빌드.
- pip 통합을 위한 PEP 503 준수 패키지 인덱스.
- CPU, CUDA, ROCm을 포함한 다양한 컴퓨팅 플랫폼 지원.
- wheel의 출처를 확보하기 위해 빌드 인증 사용.
관련
- 프로젝트
- 프로젝트
- 프로젝트
- 프로젝트
- Dispatch