Skip to content

Commit e3be364

Browse files
committed
add docker subfolder to pong notebook directly.
1 parent 90e20a6 commit e3be364

File tree

2 files changed

+78
-0
lines changed

2 files changed

+78
-0
lines changed
Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
FROM mcr.microsoft.com/azureml/openmpi3.1.2-ubuntu18.04
2+
3+
RUN pip install ray-on-aml==0.1.6
4+
RUN pip install gym[atari]==0.19.0
5+
RUN pip install gym[accept-rom-license]==0.19.0
6+
RUN pip install ale-py==0.7.0
7+
RUN pip install azureml-core
8+
RUN pip install ray==0.8.7
9+
RUN pip install ray[rllib,tune,serve]==0.8.7
10+
RUN pip install tensorflow==1.14.0
11+
12+
USER root
13+
14+
RUN apt-get update
15+
RUN apt-get install -y jq
16+
RUN apt-get install -y rsync
Lines changed: 62 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,62 @@
1+
FROM mcr.microsoft.com/azureml/openmpi4.1.0-cuda11.0.3-cudnn8-ubuntu18.04:20211111.v1
2+
3+
RUN apt-get update && apt-get install -y --no-install-recommends \
4+
python-opengl \
5+
rsync \
6+
xvfb && \
7+
apt-get clean -y && \
8+
rm -rf /var/lib/apt/lists/* && \
9+
rm -rf /usr/share/man/*
10+
11+
ENV AZUREML_CONDA_ENVIRONMENT_PATH /azureml-envs/tensorflow-2.4
12+
13+
# Create conda environment
14+
RUN conda create -p $AZUREML_CONDA_ENVIRONMENT_PATH \
15+
python=3.7 pip=20.2.4
16+
17+
# Prepend path to AzureML conda environment
18+
ENV PATH $AZUREML_CONDA_ENVIRONMENT_PATH/bin:$PATH
19+
20+
RUN pip --version
21+
RUN python --version
22+
23+
# Install ray-on-aml
24+
RUN pip install 'ray-on-aml==0.1.6'
25+
26+
RUN pip install ray==0.8.7
27+
RUN pip install gym[atari]==0.19.0
28+
RUN pip install gym[accept-rom-license]==0.19.0
29+
30+
# Install pip dependencies
31+
RUN HOROVOD_WITH_TENSORFLOW=1 \
32+
pip install 'matplotlib>=3.3,<3.4' \
33+
'psutil>=5.8,<5.9' \
34+
'tqdm>=4.59,<4.60' \
35+
'pandas>=1.1,<1.2' \
36+
'scipy>=1.5,<1.6' \
37+
'numpy>=1.10,<1.20' \
38+
'ipykernel~=6.0' \
39+
'azureml-core==1.36.0.post2' \
40+
'azureml-defaults==1.36.0' \
41+
'azureml-mlflow==1.36.0' \
42+
'azureml-telemetry==1.36.0' \
43+
'tensorboard==2.4.0' \
44+
'tensorflow-gpu==2.4.1' \
45+
'tensorflow-datasets==4.3.0' \
46+
'onnxruntime-gpu>=1.7,<1.8' \
47+
'horovod[tensorflow-gpu]==0.21.3'
48+
49+
RUN pip install --no-cache-dir \
50+
azureml-defaults \
51+
azureml-dataset-runtime[fuse,pandas] \
52+
azureml-contrib-reinforcementlearning \
53+
gputil \
54+
cloudpickle==1.3.0 \
55+
tabulate \
56+
dm_tree \
57+
lz4 \
58+
psutil \
59+
setproctitle
60+
# This is needed for mpi to locate libpython
61+
ENV LD_LIBRARY_PATH $AZUREML_CONDA_ENVIRONMENT_PATH/lib:$LD_LIBRARY_PATH
62+

0 commit comments

Comments
 (0)