安装

安装 OneFlow 稳定发布版

使用以下命令安装 OneFlow 最新的支持CUDA的稳定版本：

  1. python3 -m pip install -f https://release.oneflow.info oneflow==0.4.0+cu102

使用以下命令安装 OneFlow 最新 master 分支（不建议生产环境下使用）：

  1. python3 -m pip install oneflow -f https://staging.oneflow.info/branch/master/cu102

如果因为权限问题安装失败，请添加 --user 参数

如果提示 找不到 对应版本，请尝试升级 pip：

  1. python3 -m pip install --upgrade pip #--user

国内用户可以使用国内镜像加速

  1. python3 -m pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

详细说明可见 pypi 镜像使用帮助。

系统要求：

  • Python >= 3.6

  • CUDA 驱动要求详情见 OneFlow 源码仓库 README

从源码编译安装 OneFlow

如果你希望通过编译源码安装 OneFlow，可以参考 OneFlow 源码仓库的 README，在编译 OneFlow 源码之前，强烈推荐先阅读 Troubleshooting。

安装 OneFlow with legacy CUDA

支持其它较早版本 CUDA 的 OneFlow 的安装方法如下：

Stable:

  1. python3 -m pip install --find-links https://release.oneflow.info oneflow==0.4.0+[PLATFORM]

Nightly:

  1. python3 -m pip install oneflow -f https://staging.oneflow.info/branch/master/[PLATFORM]

其中 [PLATFORM] 可以是:

PlatformCUDA Driver VersionSupported GPUs
cu112>= 450.80.02GTX 10xx, RTX 20xx, A100, RTX 30xx
cu111>= 450.80.02GTX 10xx, RTX 20xx, A100, RTX 30xx
cu110, cu110_xla>= 450.36.06GTX 10xx, RTX 20xx, A100
cu102, cu102_xla>= 440.33GTX 10xx, RTX 20xx
cu101, cu101_xla>= 418.39GTX 10xx, RTX 20xx
cu100, cu100_xla>= 410.48GTX 10xx, RTX 20xx
cpuN/AN/A

交流QQ群

安装或使用过程遇到问题，欢迎入群与众多 OneFlow 爱好者共同讨论交流：

加QQ群 331883 或扫描二维码

qq group