FROM stablebaselines/stable-baselines-cpu

RUN apt-get -y update \
&& apt-get -y install \
ffmpeg \
freeglut3-dev \
swig \
xvfb \
&& apt-get clean \
&& rm -rf /var/lib/apt/lists/*

ENV CODE_DIR /root/code
ENV VENV /root/venv

RUN \
mkdir -p ${CODE_DIR}/rl_zoo && \
. $VENV/bin/activate && \
pip uninstall -y stable-baselines && \
pip install stable-baselines[mpi]==2.10.0 && \
pip install box2d-py==2.3.5 && \
pip install pybullet && \
pip install gym-minigrid && \
pip install scikit-optimize && \
pip install optuna && \
pip install pytablewriter && \
rm -rf $HOME/.cache/pip

ENV PATH=$VENV/bin:$PATH

RUN apt-get -y update \
&& apt-get -y install \
curl \
cmake \
ffmpeg \
freeglut3-dev \
git \
jq \
libfontconfig1 \
libglib2.0-0 \
libsm6 \
libxext6 \
libxrender1 \
libopenmpi-dev \
swig \
xvfb \
zlib1g-dev \
wget \
swig cmake libopenmpi-dev zlib1g-dev ffmpeg \
&& apt-get clean \
&& rm -rf /var/lib/apt/lists/*

# TA-Lib
RUN cd /tmp/ && \
wget http://prdownloads.sourceforge.net/ta-lib/ta-lib-0.4.0-src.tar.gz && \
tar -xvzf ta-lib-0.4.0-src.tar.gz && \
cd ta-lib/ && \
./configure --prefix=/usr && \
make && \
make install && \
#python3 setup.py install
python3 -m pip install Ta-Lib && \
cd /tmp/ && rm -rf ta-lib/

RUN python -m pip install tulipy ray[rllib]

RUN python -m pip install jhTAlib==20200211.0

CMD python -c "import jhtalib as jhta"