update libai dockerfile (Oneflow-Inc#276)

Author: L1aoXingyu

docker/Dockerfile

@@ -0,0 +1,35 @@
+FROM nvidia/cuda:11.1.1-cudnn8-devel-ubuntu18.04
+# use an older system (18.04) to avoid opencv incompatibility (issue#3524)
+
+# Comment it if you are not in China
+RUN sed -i 's/security.ubuntu.com/mirrors.ustc.edu.cn/g' /etc/apt/sources.list
+RUN sed -i 's/archive.ubuntu.com/mirrors.ustc.edu.cn/g' /etc/apt/sources.list
+
+ENV DEBIAN_FRONTEND noninteractive
+RUN apt-get update && apt-get install -y \
+	python3-opencv ca-certificates python3-dev git wget sudo ninja-build
+RUN ln -sv /usr/bin/python3 /usr/bin/python
+
+# create a non-root user
+ARG USER_ID=1000
+RUN useradd -m --no-log-init --system  --uid ${USER_ID} appuser -g sudo
+RUN echo '%sudo ALL=(ALL) NOPASSWD:ALL' >> /etc/sudoers
+USER appuser
+WORKDIR /home/appuser
+
+ENV PATH="/home/appuser/.local/bin:${PATH}"
+RUN wget https://bootstrap.pypa.io/pip/3.6/get-pip.py && \
+	python3 get-pip.py --user && \
+	rm get-pip.py
+
+# install dependencies
+# See https://pytorch.org/ for other options if you use a different version of CUDA
+RUN pip install --user pybind11 cmake   # cmake from apt-get is too old
+RUN python3 -m pip install --user --pre oneflow -f https://staging.oneflow.info/branch/master/cu112
+
+# install libai
+RUN git clone https://github.com/Oneflow-Inc/libai.git libai_repo
+
+RUN pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --user -e  libai_repo
+
+WORKDIR /home/appuser/libai_repo

docker/README.md

@@ -0,0 +1,23 @@
+## Use the container (with docker >= 19.03)
+
+```shell
+cd docker/
+# Build:
+docker build --build-arg USER_ID=$UID -t libai:v0 .
+# Launch (require GPUs):
+docker run --gpus all -it --name libai -v /home/user/:/workspace/ --ipc host --net host libai:v0 /bin/bash
+```
+
+You can also use the pre-built LiBai containers (the latest compatible version at time of publication can be pulled with `docker pull l1aoxingyu/libai:v0`).
+
+> NOTE: If your nvidia-driver cannot support `cuda111`, you can build the base image from `cuda102`.
+
+## Install new dependencies
+
+Add the following to `Dockerfile` to make persistent changes.
+
+```shell
+RUN sudo apt-get update && sudo apt-get install -y vim
+```
+
+Or run them in the container to make temporary changes.

setup.py

@@ -132,6 +132,7 @@ def get_libai_configs() -> List[str]:
             "isort==5.10.1",
             "black==21.4b ",
             "autoflake",
+            "tensorboardX",
         ],
         packages=find_packages(),
         package_data={"libai.config": get_libai_configs()},