OneFlow is a deep learning framework designed to be user-friendly, scalable and efficient. With OneFlow, it is easy to:
Bibtex Citation
@misc{yuan2021oneflow,
title={OneFlow: Redesign the Distributed Deep Learning Framework from Scratch},
author={Jinhui Yuan and Xinqi Li and Cheng Cheng and Juncheng Liu and Ran Guo and Shenghang Cai and Chi Yao and Fei Yang and Xiaodong Yi and Chuan Wu and Haoran Zhang and Jie Zhao},
year={2021},
eprint={2110.15032},
archivePrefix={arXiv},
primaryClass={cs.DC}
}
Linux. As for now, there is no pre-built release for macOS, Windows.
Python 3.7, 3.8, 3.9, 3.10
(Highly recommended) Upgrade pip
python3 -m pip install --upgrade pip #--user
CUDA Toolkit Linux x86_64 Driver
CUDA runtime is statically linked into OneFlow. OneFlow will work on a minimum supported driver, and any driver beyond. For more information, please refer to CUDA compatibility documentation.
Please upgrade your Nvidia driver to version 440.33 or above and install OneFlow for CUDA 10.2 if possible.
To install latest stable release of OneFlow with CUDA support:
python3 -m pip install oneflow
To install nightly release of OneFlow with CUDA support:
python3 -m pip install --pre oneflow -f https://staging.oneflow.info/branch/master/cu102
To install other available builds for different variants:
python3 -m pip install --find-links https://release.oneflow.info oneflow==0.8.0+[PLATFORM]
python3 -m pip install --pre oneflow -f https://staging.oneflow.info/branch/master/[PLATFORM]
All available [PLATFORM] : |
Platform | CUDA Driver Version | Supported GPUs |
---|---|---|---|
cu112 | >= 450.80.02 | GTX 10xx, RTX 20xx, A100, RTX 30xx | |
cu102 | >= 440.33 | GTX 10xx, RTX 20xx | |
cpu | N/A | N/A |
If you are in China, you could run this to have pip download packages from domestic mirror of pypi:
python3 -m pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
For more information on this, please refer to pypi 镜像使用帮助
docker pull oneflowinc/oneflow:nightly-cuda10.2
docker pull oneflowinc/oneflow:nightly-cuda11.2
Please refer to troubleshooting for common issues you might encounter when compiling and running OneFlow.
GitHub issues: any install, bug, feature issues.
www.oneflow.org: brand related information.
OneFlow was originally developed by OneFlow Inc and Zhejiang Lab.