README.md 554 Bytes
Newer Older
1
2
3
4
Dockerfile
===
## 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.
5
Dockerfile could build the customized docker image, users could build their customized docker image using this file.
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
## 2.Including Libraries  

```
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
```

## 3 How to run  
    
    docker build -t nni/nni .
    nvidia-docker run -it nni/nni