oneflow

OneFlow는 사용자 친화적이고 확장 가능하며 효율적으로 설계된 딥러닝 프레임워크입니다. 모델 프로그래밍을 위한 PyTorch 유사 API, n차원 병렬 실행으로 확장하기 위한 Global Tensor, 그리고 모델 가속 및 배포를 위한 Graph Compiler를 제공합니다. ### 주요 기능 - **PyTorch 유사 API**: PyTorch와 유사한 직관적인 모델 프로그래밍을 가능하게 합니다. - **Global Tensor**: 다차원 병렬 아키텍처 전반에 걸쳐 모델 확장을 용이하게 합니다. - **Graph Compiler**: 계산 그래프를 최적화하여 훈련을 가속화하고 배포를 간소화합니다. ### 시스템 요구 사항 - Linux 운영 체제. - Python 3.7, 3.8, 3.9, 3.10 또는 3.11. - GPU 가속을 위해 CUDA 아키텍처 6.0 이상, CUDA Toolkit 10.0 이상, Nvidia 드라이버 버전 440.33 이상. ### 설치 방법 - **Docker**: CUDA 지원이 포함된 사전 빌드된 nightly Docker 이미지를 사용할 수 있습니다. - **Pip**: 안정적인 릴리스 및 nightly 빌드(CPU 전용 또는 CUDA)는 pip를 통해 설치할 수 있으며, 중국 사용자를 위한 국내 미러 옵션도 제공됩니다. - **소스 코드**: 사용자는 저장소를 클론하고 CMake를 사용하여 빌드할 수 있으며, CPU 전용 또는 CUDA 환경을 위한 구성 사전 설정이 제공됩니다. ### 생태계 및 모델 저장소 - **Libai**: BERT, GPT, T5, VisionTransformer 및 SwinTransformer를 포함한 대규모 트랜스포머 모델의 병렬 훈련을 위해 설계된 도구 상자입니다. - **FlowVision**: 컴퓨터 비전 데이터 세트, 최신 모델 및 유틸리티를 제공하는 도구 상자입니다. - **OneFlow-Models**: ResNet-50 및 Wide&Deep과 같은 모델 구현을 제공합니다(구식으로 표시됨). ### 커뮤니티 및 커뮤니케이션 - **중국 채널**: QQ 그룹(331883), WeChat(OneFlowXZS) 및 Zhihu 조직 페이지. - **국제 채널**: Discord 서버, Twitter, LinkedIn 및 Medium 블로그. - 버그 보고 및 기능 요청은 GitHub 이슈를 통해 관리됩니다. OneFlow는 원래 OneFlow Inc.와 Zhejiang Lab에서 개발되었으며 Apache License 2.0에 따라 배포됩니다.