NNI (Neural Network Intelligence) is a toolkit to help users run automated machine learning (AutoML) experiments.
NNI (Neural Network Intelligence) is a toolkit to help users run automated machine learning (AutoML) experiments.
The tool dispatches and runs trial jobs generated by tuning algorithms to search the best neural architecture and/or hyper-parameters in different environments like local machine, remote servers and cloud.
The tool dispatches and runs trial jobs generated by tuning algorithms to search the best neural architecture and/or hyper-parameters in different environments like local machine, remote servers and cloud.
### **NNI [v0.5](https://github.com/Microsoft/nni/releases) has been released!**
### **NNI [v0.5.1](https://github.com/Microsoft/nni/releases) has been released!**
@@ -6,13 +6,13 @@ This is the Dockerfile of NNI project. It includes serveral popular deep learnin
...
@@ -6,13 +6,13 @@ This is the Dockerfile of NNI project. It includes serveral popular deep learnin
```
```
CUDA 9.0, CuDNN 7.0
CUDA 9.0, CuDNN 7.0
numpy 1.14.3,scipy 1.1.0
numpy 1.14.3,scipy 1.1.0
TensorFlow 1.10.0
TensorFlow-gpu 1.10.0
Keras 2.1.6
Keras 2.1.6
PyTorch 0.4.1
PyTorch 0.4.1
scikit-learn 0.20.0
scikit-learn 0.20.0
pandas 0.23.4
pandas 0.23.4
lightgbm 2.2.2
lightgbm 2.2.2
NNI v0.5
NNI v0.5.1
```
```
You can take this Dockerfile as a reference for your own customized Dockerfile.
You can take this Dockerfile as a reference for your own customized Dockerfile.
...
@@ -26,6 +26,8 @@ __Run the docker image__
...
@@ -26,6 +26,8 @@ __Run the docker image__
```
```
docker run -it nni/nni
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](https://github.com/NVIDIA/nvidia-docker), then run the following command
* 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