fix docker in aml (#2633)

microsoft · Jul 6, 2020 · 9c4a1c7 · 9c4a1c7
1 parent 66fec2c
commit 9c4a1c7
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 4 deletions.
diff --git a/deployment/docker/Dockerfile b/deployment/docker/Dockerfile
@@ -29,6 +29,11 @@ RUN DEBIAN_FRONTEND=noninteractive && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*
 
+#
+# generate python script
+#
+RUN cp /usr/bin/python3 /usr/bin/python
+
 #
 # update pip
 #
@@ -69,6 +74,13 @@ RUN python3 -m pip --no-cache-dir install pandas==0.23.4 lightgbm==2.2.2
 #
 RUN python3 -m pip --no-cache-dir install nni
 
+#
+# install aml package
+#
+RUN python3 -m pip --no-cache-dir install azureml
+RUN python3 -m pip --no-cache-dir install azureml-sdk
+
+
 ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/root/.local/bin:/usr/bin:/bin:/sbin
 
 WORKDIR /root
diff --git a/src/nni_manager/config/aml/amlUtil.py b/src/nni_manager/config/aml/amlUtil.py
@@ -28,10 +28,7 @@
     compute_target = ComputeTarget(workspace=ws, name=args.compute_target)
     experiment = Experiment(ws, args.experiment_name)
     run_config = RunConfiguration()
-    dependencies = CondaDependencies()
-    dependencies.add_pip_package("azureml-sdk")
-    dependencies.add_pip_package("azureml")
-    run_config.environment.python.conda_dependencies = dependencies
+    run_config.environment.python.user_managed_dependencies = True
     run_config.environment.docker.enabled = True
     run_config.environment.docker.base_image = args.docker_image
     run_config.target = compute_target