diff --git a/deployment/docker/README.md b/deployment/docker/README.md index f15b9ddd6a..aa8dae38a4 100644 --- a/deployment/docker/README.md +++ b/deployment/docker/README.md @@ -4,15 +4,17 @@ Dockerfile 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 +CUDA 9.0 +CuDNN 7.0 +numpy 1.14.3 +scipy 1.1.0 +tensorflow-gpu 1.15.0 +keras 2.1.6 +torch 1.4.0 scikit-learn 0.20.0 pandas 0.23.4 lightgbm 2.2.2 -NNI v0.7 +nni ``` You can take this Dockerfile as a reference for your own customized Dockerfile. diff --git a/docs/en_US/TrainingService/AMLMode.md b/docs/en_US/TrainingService/AMLMode.md index b77da982af..0907f29a02 100644 --- a/docs/en_US/TrainingService/AMLMode.md +++ b/docs/en_US/TrainingService/AMLMode.md @@ -22,8 +22,8 @@ Step 6. Create an AML cluster as the computeTarget. Step 7. Open a command line and install AML package environment. ``` -python3 -m pip install azureml --user -python3 -m pip install azureml-sdk --user +python3 -m pip install azureml +python3 -m pip install azureml-sdk ``` ## Run an experiment