mirror of
https://github.com/Azure/MachineLearningNotebooks.git
synced 2025-12-19 17:17:04 -05:00
56 lines
1.5 KiB
Plaintext
56 lines
1.5 KiB
Plaintext
# NC-series GPUs only support the pytorch-1.11/cuda11.3 combo
|
|
# See https://learn.microsoft.com/en-us/azure/machine-learning/resource-curated-environments
|
|
FROM mcr.microsoft.com/azureml/curated/acpt-pytorch-1.11-cuda11.3
|
|
|
|
USER root
|
|
|
|
# ENV AZUREML_CONDA_ENVIRONMENT_PATH /azureml-envs/ray-rllib
|
|
# Create conda environment
|
|
# RUN conda create -p $AZUREML_CONDA_ENVIRONMENT_PATH \
|
|
# python=3.8.5
|
|
|
|
# Prepend path to AzureML conda environment
|
|
# ENV PATH $AZUREML_CONDA_ENVIRONMENT_PATH/bin:$PATH
|
|
|
|
# Install necessary packages to support videos in rllib/gym
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
python-opengl \
|
|
rsync \
|
|
xvfb && \
|
|
apt-get clean -y && \
|
|
rm -rf /var/lib/apt/lists/* && \
|
|
rm -rf /usr/share/man/*
|
|
|
|
RUN pip --version
|
|
RUN python --version
|
|
|
|
RUN pip install ray-on-aml==0.2.4 \
|
|
ray==2.4.0 \
|
|
ray[rllib]==2.4.0 \
|
|
mlflow==2.3.1 \
|
|
azureml-defaults==1.50.0 \
|
|
azureml-dataset-runtime[fuse,pandas]==1.50.0 \
|
|
azureml-contrib-reinforcementlearning==1.50.0 \
|
|
gputil==1.4.0 \
|
|
scipy==1.9.1 \
|
|
pyglet==2.0.6 \
|
|
cloudpickle==2.2.1 \
|
|
tensorflow==2.11.0 \
|
|
tensorflow-probability==0.19.0 \
|
|
tabulate==0.9.0 \
|
|
dm_tree==0.1.8 \
|
|
lz4==4.3.2 \
|
|
psutil==5.9.4 \
|
|
setproctitle==1.3.2 \
|
|
pygame==2.1.0 \
|
|
gymnasium[classic_control]==0.26.3 \
|
|
gymnasium[atari]==0.26.3 \
|
|
gymnasium[accept-rom-license]==0.26.3 \
|
|
gym==0.26.2 \
|
|
gym[atari]==0.26.2 \
|
|
gym[accept-rom-license]==0.26.2
|
|
|
|
# Display all versions
|
|
RUN pip freeze
|
|
|