开源软件名称:OneFlow
开源软件地址:https://gitee.com/mirrors/OneFlow
开源软件介绍:
OneFlowOneFlow is a performance-centered and open-source deep learning framework.
Latest News- Version 0.7.0 is out!
- Introducing global tensor
- Semi-auto parallelization has landed
- Full changelog
PublicationInstall OneFlowSystem RequirementsPython 3.6, 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.
Install with Pip PackageTo install latest stable release of OneFlow with CUDA support: python3 -m pip install -f https://release.oneflow.info oneflow==0.7.0+cu102 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: - Stable
python3 -m pip install --find-links https://release.oneflow.info oneflow==0.7.0+[PLATFORM] - Nightly
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 镜像使用帮助
Use docker imagedocker pull oneflowinc/oneflow:nightly-cuda10.2docker pull oneflowinc/oneflow:nightly-cuda11.2 Build from SourceClone Source CodeBuild OneFlowTroubleshootingPlease refer to troubleshooting for common issues you might encounter when compiling and running OneFlow. Advanced featuresXRT- You can check this doc to obtain more details about how to use XLA and TensorRT with OneFlow.
Getting StartedDocumentationModel Zoo and BenchmarkCommunicationThe TeamOneFlow was originally developed by OneFlow Inc and Zhejiang Lab. LicenseApache License 2.0 |
请发表评论