Skip to content
This repository has been archived by the owner on Sep 18, 2024. It is now read-only.

update doc about nni docker image #345

Merged
merged 1 commit into from
Nov 10, 2018
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
27 changes: 20 additions & 7 deletions deployment/docker/README.md
Original file line number Diff line number Diff line change
@@ -1,20 +1,33 @@
Dockerfile
===
## 1.Description
## 1.Description
This is the Dockerfile of nni project, including the most kinds of deeplearning frameworks and nni source code. You can run your nni experiment in this docker container directly.
Dockerfile could build the customized docker image, users could build their customized docker image using this file.
## 2.Including Libraries
This docker file includes the following libraries on `Ubuntu 16.04 LTS`:

```
Ubuntu 16.04 LTS
CUDA 9.0, CuDNN 7.0
numpy 1.14.3,scipy 1.1.0
TensorFlow 1.5.0
Keras 2.1.6
NNI v0.1
NNI v0.3
```

## 3 How to run

## 2.How to build and run
__Use the following command to build docker image__
```
docker build -t nni/nni .
nvidia-docker run -it nni/nni
```
__Run the docker image__
* If does not use GPU in docker container, simply run the following command
```
docker run -it nni/nni
```
* If use GPU in docker container, make sure you have installed [NVIDIA Container Runtime](https://github.com/NVIDIA/nvidia-docker), then run the following command
```
nvidia-docker run -it nni/nni
```
or
```
docker run --runtime=nvidia -it nni/nni
```
4 changes: 4 additions & 0 deletions docs/InstallNNI_Ubuntu.md
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,10 @@
cd nni
source install.sh

* __Install NNI in docker image__

You can also install NNI in a docker image. Please follow the instructions [here](../deployment/docker/README.md) to build NNI docker image.

## Further reading
* [Overview](Overview.md)
* [Use command line tool nnictl](NNICTLDOC.md)
Expand Down