feat: Updated dockerfile.

instadeepai · arnupretorius · Dec 2, 2021 · Aug 4, 2021 · Aug 5, 2021 · Aug 23, 2021
commit 9c7a9e62614691cfc2501f2ea90b7cfe870a4f04
@@ -0,0 +1,90 @@
+##########################################################
+# Core Mava image
+FROM nvcr.io/nvidia/cuda:11.4.2-cudnn8-runtime-ubuntu20.04
+# Update packages
+RUN apt-get update -y && apt-get install -y python3-pip
+# Update python path
+RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.8 10 &&\
+    rm -rf /root/.cache && apt-get clean
+# Upgrade pip
+RUN python -m pip install --upgrade pip
+# Location of mava folder
+ARG folder=/home/app/mava
+# Flag to record agents
+ARG record
+## working directory
+WORKDIR ${folder}
+## Copy code from current path.
+COPY . /home/app/mava
+# For box2d
+RUN apt-get install swig -y
+## Install core dependencies.
+RUN python -m pip install -e .[reverb,launchpad]
+## Optional install for screen recording.
+ENV DISPLAY=:0
+RUN if [ "$record" = "true" ]; then \
+    ./bash_scripts/install_record.sh; \
+fi
+EXPOSE 6006
+##########################################################
+
+##########################################################
+# Core Mava-TF image
+FROM mava-core as tf-core
+# Tensorflow gpu config.
+ENV TF_FORCE_GPU_ALLOW_GROWTH=true
+ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
+ENV TF_CPP_MIN_LOG_LEVEL=3
+## Install core tf dependencies.
+RUN python -m pip install -e .[tf]
+##########################################################
+
+##########################################################
+# PZ image
+FROM tf-core AS pz
+RUN python -m pip install -e .[pz]
+# PettingZoo Atari envs
+RUN apt-get update
+RUN apt-get install ffmpeg libsm6 libxext6  -y
+RUN apt-get install -y unrar-free
+RUN python -m pip install autorom
+RUN AutoROM -v
+##########################################################
+
+##########################################################
+# SMAC image
+FROM tf-core AS sc2
+## Install smac environment
+RUN apt-get -y install git
+RUN pip install .[sc2]
+# We use the pz wrapper for smac
+RUN pip install .[pz]
+ENV SC2PATH /home/app/mava/3rdparty/StarCraftII
+##########################################################
+
+##########################################################
+# Flatland Image
+FROM tf-core AS flatland
+RUN python -m pip install -e .[flatland]
+##########################################################
+
+##########################################################
+## Robocup Image
+FROM tf-core AS robocup
+RUN ./bash_scripts/install_robocup.sh
+##########################################################
+
+##########################################################
+## OpenSpiel Image
+FROM tf-core AS openspiel
+RUN pip install .[open_spiel]
+##########################################################
+
+##########################################################
+# MeltingPot Image
+FROM tf-core AS meltingpot
+# Install meltingpot
+RUN ./bash_scripts/install_meltingpot.sh
+# Add meltingpot to python path
+ENV PYTHONPATH "${PYTHONPATH}:${folder}/meltingpot"
+##########################################################
@@ -44,7 +44,7 @@ endif
 IMAGE = $(DOCKER_IMAGE_NAME):$(DOCKER_IMAGE_TAG)
 # make file commands
 build:
-	DOCKER_BUILDKIT=1 docker build --tag $(IMAGE) -f Dockerfile.tf --target $(DOCKER_IMAGE_TAG)  --build-arg record=$(record) .
+	DOCKER_BUILDKIT=1 docker build --tag $(IMAGE) --target $(DOCKER_IMAGE_TAG)  --build-arg record=$(record) .
 
 run:
 	$(DOCKER_RUN) python $(example) --base_dir /home/app/mava/logs/