mirror of
https://github.com/Azure/MachineLearningNotebooks.git
synced 2025-12-20 09:37:04 -05:00
79 lines
2.0 KiB
Docker
79 lines
2.0 KiB
Docker
FROM mcr.microsoft.com/azureml/base-gpu:openmpi3.1.2-cuda10.0-cudnn7-ubuntu18.04
|
|
|
|
# Install some basic utilities
|
|
RUN apt-get update && apt-get install -y \
|
|
curl \
|
|
ca-certificates \
|
|
sudo \
|
|
cpio \
|
|
git \
|
|
bzip2 \
|
|
libx11-6 \
|
|
tmux \
|
|
htop \
|
|
gcc \
|
|
xvfb \
|
|
python-opengl \
|
|
x11-xserver-utils \
|
|
ffmpeg \
|
|
mesa-utils \
|
|
nano \
|
|
vim \
|
|
rsync \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Create a working directory
|
|
RUN mkdir /app
|
|
WORKDIR /app
|
|
|
|
# Create a Python 3.7 environment
|
|
RUN conda install conda-build \
|
|
&& conda create -y --name py37 python=3.7.3 \
|
|
&& conda clean -ya
|
|
ENV CONDA_DEFAULT_ENV=py37
|
|
|
|
# Install Minecraft needed libraries
|
|
RUN mkdir -p /usr/share/man/man1 && \
|
|
sudo apt-get update && \
|
|
sudo apt-get install -y \
|
|
openjdk-8-jre-headless=8u162-b12-1 \
|
|
openjdk-8-jdk-headless=8u162-b12-1 \
|
|
openjdk-8-jre=8u162-b12-1 \
|
|
openjdk-8-jdk=8u162-b12-1
|
|
|
|
RUN pip install --upgrade --user minerl
|
|
|
|
# PyTorch with CUDA 10 installation
|
|
RUN conda install -y -c pytorch \
|
|
cuda100=1.0 \
|
|
magma-cuda100=2.4.0 \
|
|
"pytorch=1.1.0=py3.7_cuda10.0.130_cudnn7.5.1_0" \
|
|
torchvision=0.3.0 \
|
|
&& conda clean -ya
|
|
|
|
RUN pip install \
|
|
pandas \
|
|
matplotlib \
|
|
numpy \
|
|
scipy \
|
|
azureml-defaults \
|
|
tensorboardX \
|
|
tensorflow-gpu==1.15rc2 \
|
|
GPUtil \
|
|
tabulate \
|
|
dm_tree \
|
|
lz4 \
|
|
ray==0.8.3 \
|
|
ray[rllib]==0.8.3 \
|
|
ray[tune]==0.8.3
|
|
|
|
COPY patch_files/* /root/.local/lib/python3.7/site-packages/minerl/env/Malmo/Minecraft/src/main/java/com/microsoft/Malmo/Client/
|
|
|
|
# Start minerl to pre-fetch minerl files (saves time when starting minerl during training)
|
|
RUN xvfb-run -a -s "-screen 0 1400x900x24" python -c "import gym; import minerl; env = gym.make('MineRLTreechop-v0'); env.close();"
|
|
|
|
RUN pip install --index-url https://test.pypi.org/simple/ malmo && \
|
|
python -c "import malmo.minecraftbootstrap; malmo.minecraftbootstrap.download();"
|
|
|
|
ENV MALMO_XSD_PATH="/app/MalmoPlatform/Schemas"
|