Docker Support

[aswinkumar1999]

I noticed that the repository did not have Dockerfile to run it with Docker. Are there any specific reasoning for it?

I was able to test a couple of commands with this Dockerfile, I could create a PR if it's there are no specific reasons to use a Dockerfile.

# Build : docker build -t mujoco_playground_jax .
# Run : xhost +local:docker && docker run --rm -it --gpus all --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 -e DISPLAY=$DISPLAY -v /tmp/.X11-unix:/tmp/.X11-unix -v "$HOME/.Xauthority:/root/.Xauthority:rw" mujoco_playground_jax bash
# Users can navigate to /app/mujoco_playground/examples to run scripts.

# Use the specified NVIDIA JAX base image which includes Python, JAX, and CUDA.
FROM nvcr.io/nvidia/jax:25.01-py3

# Set the working directory inside the container
WORKDIR /app

# Install system dependencies and git.
RUN apt-get update && \
    apt-get install -y --no-install-recommends \
    libgl1 \
    libglx-mesa0 \
    libglew-dev \
    libosmesa6 \
    libglfw3 \
    patchelf \
    ffmpeg \
    git && \
    apt-get clean && \
    rm -rf /var/lib/apt/lists/*

# Clone the mujoco_playground repository into a subdirectory [2, 3, 4, 5].
RUN git clone https://github.com/google-deepmind/mujoco_playground.git ./mujoco_playground

# Upgrade pip and install the necessary Python packages
RUN pip install --no-cache-dir --upgrade pip && \
    pip install --no-cache-dir mediapy wandb \
    pip install --no-cache-dir mujoco playground

# Set the default command to launch a bash shell for interactive use.
ENV MUJOCO_GL=egl
CMD ["bash"]

[btaba]

Hi @aswinkumar1999 , we did not include Dockerfiles because all of the deps can be pip installed in a python virtual env (>=3.10), and JAX has good support across hardware platforms. We're not really looking to maintain Dockerfiles right now, but we appreciate you sharing a minimal Dockerfile!

[aswinkumar1999]

Sounds great!

I have replaced the above Dockerfile with a newer one with some fixes.