Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
nni
Commits
9c4a1c74
Unverified
Commit
9c4a1c74
authored
Jul 06, 2020
by
SparkSnail
Committed by
GitHub
Jul 06, 2020
Browse files
fix docker in aml (#2633)
parent
66fec2c8
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
13 additions
and
4 deletions
+13
-4
deployment/docker/Dockerfile
deployment/docker/Dockerfile
+12
-0
src/nni_manager/config/aml/amlUtil.py
src/nni_manager/config/aml/amlUtil.py
+1
-4
No files found.
deployment/docker/Dockerfile
View file @
9c4a1c74
...
@@ -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
src/nni_manager/config/aml/amlUtil.py
View file @
9c4a1c74
...
@@ -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
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment