FROM python:3.8-slim

# Install gcc & g++
RUN apt-get update && apt-get install -y gcc g++

# Download Mujoco 210
RUN apt-get update && apt-get install -y wget libgl1-mesa-glx
RUN wget XXXX && \
    mkdir -p /root/.mujoco && \
    tar -zxvf mujoco210-linux-x86_64.tar.gz -C /root/.mujoco && \
    rm mujoco210-linux-x86_64.tar.gz
ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/root/.mujoco/mujoco210/bin

# Install git
RUN apt-get update && apt-get install -y git

# Install required Python packages
ADD requirements.txt /tmp/requirements.txt
RUN pip install setuptools==63.2.0 && \
    pip install wheel==0.38.4 && \
    apt-get install -y swig && \
    apt install -y libosmesa6-dev && \
    pip install -r /tmp/requirements.txt && \
    rm /tmp/requirements.txt
RUN pip install torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 --index-url XXXX
