oneflow-101

OneFlow Tutorial

OneFlow介绍

• 官网：OneFlow
• Github: OneFlow
• 文档：OneFlow 文档

OneFlow 101

base

• mlp_mnist
• lenet_mnist

安装环境依赖

此项目依赖docker和nvidia-docker，因nvidia-docker只有Linux版本，默认系统为Linux

• docker: docker install
• nvidia-docker: nvidia-docker install

如果在window平台使用wsl2，可参考wsl2 install nvidia-docker

运行oneflow-101

创建镜像

1. git clone https://github.com/xxg1413/oneflow-101

2. cd oneflow-101 && docker build -t oneflow .

运行

nvidia-docker run -p 8888:8888 oneflow

可以看到jupyter已经启动，把 127.0.0.1 替换为 本机IP，在浏览器中打开

如果使用 nvidia-docker run -d -p 8888:8888 oneflow

则使用docker logs 容器ID 查看日志，把 127.0.0.1 替换为 本机IP，在浏览器中打开