FROM tensorflow/tensorflow:2.4.1-gpu

WORKDIR /home

RUN pip install --upgrade pip

RUN apt-get -y update
RUN apt-get -y install git
RUN apt-get -y install graphviz

COPY requirements.txt requirements.txt

RUN pip install -r requirements.txt


CMD bash
