FROM tensorflow/tensorflow:latest-gpu-py3-jupyter
RUN apt-get update
RUN apt-get install -y \
    git \
    wget 
# OpenCV dependencies
RUN apt-get install -y \
    libsm6 \
    libxext6 \
    libxrender1
# OpenCV package
RUN pip install opencv-python \
    opencv-contrib-python
# gRPC package
RUN pip install grpcio \
    grpcio-tools
RUN pip install Pillow
RUN apt-get install unzip
WORKDIR /home
# Protobuf Install
RUN wget -O protobuf.zip https://github.com/protocolbuffers/protobuf/releases/download/v3.7.1/protoc-3.7.1-linux-x86_64.zip && \
    unzip protobuf.zip -d "/home/protobuf"
RUN rm -rf protobuf.zip
# Punya opencv
RUN apt-get install -y libglib2.0-0
RUN pip install --upgrade pip
RUN pip install keras
RUN pip install wget
RUN pip install seaborn
# WORKDIR /home
CMD bash