mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-02 06:16:32 +00:00
34 lines
1.3 KiB
Docker
34 lines
1.3 KiB
Docker
# A Dockerfile that sets up a full Gymnasium install with test dependencies
|
|
ARG PYTHON_VERSION
|
|
FROM python:$PYTHON_VERSION
|
|
|
|
SHELL ["/bin/bash", "-o", "pipefail", "-c"]
|
|
|
|
RUN apt-get -y update \
|
|
&& apt-get install --no-install-recommends -y \
|
|
libglu1-mesa-dev libgl1-mesa-dev libosmesa6-dev \
|
|
xvfb unzip patchelf ffmpeg cmake swig \
|
|
&& apt-get autoremove -y \
|
|
&& apt-get clean \
|
|
&& rm -rf /var/lib/apt/lists/* \
|
|
# Download mujoco
|
|
&& mkdir /root/.mujoco \
|
|
&& cd /root/.mujoco \
|
|
&& wget -qO- 'https://github.com/deepmind/mujoco/releases/download/2.1.0/mujoco210-linux-x86_64.tar.gz' | tar -xzvf -
|
|
|
|
ENV LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/root/.mujoco/mujoco210/bin"
|
|
|
|
# Build mujoco-py from source. Pypi installs wheel packages and Cython won't recompile old file versions in the Github Actions CI.
|
|
# Thus generating the following error https://github.com/cython/cython/pull/4428
|
|
RUN git clone https://github.com/openai/mujoco-py.git\
|
|
&& cd mujoco-py \
|
|
&& pip install -e .
|
|
|
|
COPY . /usr/local/gymnasium/
|
|
WORKDIR /usr/local/gymnasium/
|
|
|
|
# Test with PyTorch CPU build, since CUDA is not available in CI anyway
|
|
RUN pip install .[all,testing] --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cpu
|
|
|
|
ENTRYPOINT ["/usr/local/gymnasium/bin/docker_entrypoint"]
|