Unverified Commit 9c4a1c74 authored by SparkSnail's avatar SparkSnail Committed by GitHub
Browse files

fix docker in aml (#2633)

parent 66fec2c8
...@@ -29,6 +29,11 @@ RUN DEBIAN_FRONTEND=noninteractive && \ ...@@ -29,6 +29,11 @@ RUN DEBIAN_FRONTEND=noninteractive && \
apt-get clean && \ apt-get clean && \
rm -rf /var/lib/apt/lists/* rm -rf /var/lib/apt/lists/*
#
# generate python script
#
RUN cp /usr/bin/python3 /usr/bin/python
# #
# update pip # update pip
# #
...@@ -69,6 +74,13 @@ RUN python3 -m pip --no-cache-dir install pandas==0.23.4 lightgbm==2.2.2 ...@@ -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 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 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 WORKDIR /root
...@@ -28,10 +28,7 @@ if __name__ == "__main__": ...@@ -28,10 +28,7 @@ if __name__ == "__main__":
compute_target = ComputeTarget(workspace=ws, name=args.compute_target) compute_target = ComputeTarget(workspace=ws, name=args.compute_target)
experiment = Experiment(ws, args.experiment_name) experiment = Experiment(ws, args.experiment_name)
run_config = RunConfiguration() run_config = RunConfiguration()
dependencies = CondaDependencies() run_config.environment.python.user_managed_dependencies = True
dependencies.add_pip_package("azureml-sdk")
dependencies.add_pip_package("azureml")
run_config.environment.python.conda_dependencies = dependencies
run_config.environment.docker.enabled = True run_config.environment.docker.enabled = True
run_config.environment.docker.base_image = args.docker_image run_config.environment.docker.base_image = args.docker_image
run_config.target = compute_target run_config.target = compute_target
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment