mirror of
https://github.com/saymrwulf/stable-baselines3.git
synced 2026-05-14 20:58:03 +00:00
27 lines
971 B
Docker
27 lines
971 B
Docker
ARG PARENT_IMAGE
|
|
FROM $PARENT_IMAGE
|
|
ARG PYTORCH_DEPS=cpuonly
|
|
ARG PYTHON_VERSION=3.12
|
|
ARG MAMBA_DOCKERFILE_ACTIVATE=1 # (otherwise python will not be found)
|
|
|
|
# Install micromamba env and dependencies
|
|
RUN micromamba install -n base -y python=$PYTHON_VERSION \
|
|
pytorch $PYTORCH_DEPS -c conda-forge -c pytorch -c nvidia && \
|
|
micromamba clean --all --yes
|
|
|
|
ENV CODE_DIR=/home/$MAMBA_USER
|
|
|
|
# Copy setup file only to install dependencies
|
|
COPY --chown=$MAMBA_USER:$MAMBA_USER ./setup.py ${CODE_DIR}/stable-baselines3/setup.py
|
|
COPY --chown=$MAMBA_USER:$MAMBA_USER ./stable_baselines3/version.txt ${CODE_DIR}/stable-baselines3/stable_baselines3/version.txt
|
|
|
|
RUN cd ${CODE_DIR}/stable-baselines3 && \
|
|
pip install uv && \
|
|
uv pip install --system -e .[extra,tests,docs] && \
|
|
# Use headless version for docker
|
|
uv pip uninstall opencv-python && \
|
|
uv pip install --system opencv-python-headless && \
|
|
pip cache purge && \
|
|
uv cache clean
|
|
|
|
CMD /bin/bash
|