mirror of
https://github.com/saymrwulf/stable-baselines3.git
synced 2026-05-14 20:58:03 +00:00
c62e9259db
* Add support for custom objects * Add python 3.8 to the CI * Bump version * PyType fixes * [ci skip] Fix typo * Add note about slow-down + fix typos * Minor edits to the doc * Bug fix for DQN * Update test * Add test for custom objects
41 lines
1.3 KiB
Docker
41 lines
1.3 KiB
Docker
ARG PARENT_IMAGE
|
|
FROM $PARENT_IMAGE
|
|
ARG PYTORCH_DEPS=cpuonly
|
|
ARG PYTHON_VERSION=3.6
|
|
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
build-essential \
|
|
cmake \
|
|
git \
|
|
curl \
|
|
ca-certificates \
|
|
libjpeg-dev \
|
|
libpng-dev \
|
|
libglib2.0-0 && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
# Install Anaconda and dependencies
|
|
RUN curl -o ~/miniconda.sh https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh && \
|
|
chmod +x ~/miniconda.sh && \
|
|
~/miniconda.sh -b -p /opt/conda && \
|
|
rm ~/miniconda.sh && \
|
|
/opt/conda/bin/conda install -y python=$PYTHON_VERSION numpy pyyaml scipy ipython mkl mkl-include && \
|
|
/opt/conda/bin/conda install -y pytorch $PYTORCH_DEPS -c pytorch && \
|
|
/opt/conda/bin/conda clean -ya
|
|
ENV PATH /opt/conda/bin:$PATH
|
|
|
|
ENV CODE_DIR /root/code
|
|
|
|
# Copy setup file only to install dependencies
|
|
COPY ./setup.py ${CODE_DIR}/stable-baselines3/setup.py
|
|
COPY ./stable_baselines3/version.txt ${CODE_DIR}/stable-baselines3/stable_baselines3/version.txt
|
|
|
|
RUN \
|
|
cd ${CODE_DIR}/stable-baselines3 3&& \
|
|
pip install -e .[extra,tests,docs] && \
|
|
# Use headless version for docker
|
|
pip uninstall -y opencv-python && \
|
|
pip install opencv-python-headless && \
|
|
rm -rf $HOME/.cache/pip
|
|
|
|
CMD /bin/bash
|