|
| 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