60 lines
2.2 KiB
Docker
60 lines
2.2 KiB
Docker
##########################################
|
|
# Dockerfile for TD-MPC2 #
|
|
# TD-MPC2 Anonymous Authors, 2023 (c) #
|
|
# -------------------------------------- #
|
|
# Instructions: #
|
|
# docker build . -t <user>/tdmpc2:0.1.0 #
|
|
# docker push <user>/tdmpc2:0.1.0 #
|
|
##########################################
|
|
|
|
# base image
|
|
FROM nvidia/cudagl:11.3.1-devel-ubuntu20.04
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
|
|
# packages
|
|
RUN apt-get -y update && \
|
|
apt-get install -y --no-install-recommends build-essential git nano rsync vim tree curl \
|
|
wget unzip htop tmux xvfb patchelf ca-certificates bash-completion libjpeg-dev libpng-dev \
|
|
ffmpeg cmake swig libssl-dev libcurl4-openssl-dev libopenmpi-dev python3-dev zlib1g-dev \
|
|
qtbase5-dev qtdeclarative5-dev libglib2.0-0 libglu1-mesa-dev libgl1-mesa-dev libvulkan1 \
|
|
libgl1-mesa-glx libosmesa6 libosmesa6-dev libglew-dev mesa-utils && \
|
|
apt-get clean && \
|
|
apt-get autoremove -y && \
|
|
rm -rf /var/lib/apt/lists/* && \
|
|
mkdir /root/.ssh
|
|
|
|
# miniconda
|
|
RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda.sh && \
|
|
/bin/bash ~/miniconda.sh -b -p /opt/conda && \
|
|
rm ~/miniconda.sh && \
|
|
. /opt/conda/etc/profile.d/conda.sh && \
|
|
conda init && \
|
|
conda clean -ya
|
|
ENV PATH /opt/conda/bin:$PATH
|
|
SHELL ["/bin/bash", "-c"]
|
|
|
|
# conda environment
|
|
COPY nvidia_icd.json /usr/share/vulkan/icd.d/nvidia_icd.json
|
|
COPY environment.yaml /root
|
|
RUN conda env update -n base -f /root/environment.yaml && \
|
|
rm /root/environment.yaml && \
|
|
cd /root && \
|
|
python -m mani_skill2.utils.download_asset all -y && \
|
|
conda clean -ya && \
|
|
pip cache purge
|
|
|
|
# environment variables
|
|
ENV MUJOCO_GL egl
|
|
ENV MS2_ASSET_DIR /root/data
|
|
ENV LD_LIBRARY_PATH /root/.mujoco/mujoco210/bin:${LD_LIBRARY_PATH}
|
|
|
|
# mujoco (required for metaworld)
|
|
RUN mkdir -p /root/.mujoco && \
|
|
wget https://www.tdmpc2.com/files/mjkey.txt && \
|
|
wget https://github.com/deepmind/mujoco/releases/download/2.1.0/mujoco210-linux-x86_64.tar.gz && \
|
|
tar -xzf mujoco210-linux-x86_64.tar.gz && \
|
|
rm mujoco210-linux-x86_64.tar.gz && \
|
|
mv mujoco210 /root/.mujoco/mujoco210 && \
|
|
mv mjkey.txt /root/.mujoco/mjkey.txt && \
|
|
python -c "import mujoco_py"
|