FROM nvidia/cuda:11.0.3-cudnn8-runtime-ubuntu18.04 AS python-ubuntu

RUN apt-get update && \
    apt-get install -y build-essential curl python3 python3-pip python3-venv && \
    rm -rf /var/lib/apt/lists/*

# Inorder to run pip3.6, run it the following way `python3.6 -m pip`
# Update Python & Install wheel
RUN python3 -m pip install pip --upgrade && python3 -m pip install wheel

FROM python-ubuntu

RUN apt-get update && \
    apt-get install liblapack-dev libpng-dev libfreetype6-dev libqhull-dev git pkg-config portaudio19-dev swig libpulse-ocaml-dev gfortran libopenblas-dev libatlas-base-dev -y && \
    rm -rf /var/lib/apt/lists/*

RUN mkdir /training && pip3 install git+https://github.com/tystuyfzand/precise-lite.git@feature/tf24 && pip3 install tensorflow-gpu==2.4.4

WORKDIR /training

COPY entrypoint.sh /entrypoint.sh

ENTRYPOINT ["/entrypoint.sh"]
CMD ["/bin/bash"]