Dockerfile

1.Description

This is the Dockerfile of NNI project. It includes serveral popular deep learning frameworks and NNI. It is tested on Ubuntu 16.04 LTS:

CUDA 9.0, CuDNN 7.0
numpy 1.14.3,scipy 1.1.0
TensorFlow-gpu 1.10.0
Keras 2.1.6
PyTorch 0.4.1
scikit-learn 0.20.0
pandas 0.23.4
lightgbm 2.2.2
NNI v0.7

You can take this Dockerfile as a reference for your own customized Dockerfile.

2.How to build and run

Use the following command from nni/deployment/docker to build docker image

    docker build -t nni/nni .

Run the docker image

• If does not use GPU in docker container, simply run the following command

    docker run -it nni/nni

Note that if you want to use tensorflow, please uninstall tensorflow-gpu and install tensorflow in this docker container. Or modify Dockerfile to install tensorflow (without gpu) and build docker image.

• If use GPU in docker container, make sure you have installed NVIDIA Container Runtime, then run the following command

    nvidia-docker run -it nni/nni

or

    docker run --runtime=nvidia -it nni/nni

3.Directly retrieve the docker image

Use the following command to retrieve the NNI docker image from Docker Hub

docker pull msranni/nni:latest