FROM nvcr.io/nvidia/pytorch:20.03-py3 
#FROM flare-detection-pytorch:v2.0 
RUN rm -rf /var/lib/apt/lists/* && rm -rf /root/.cache/pip/ 
RUN apt-get update 
RUN pip uninstall torch torchvision -y 
RUN apt-get install -y libsm6 libxext6 libxrender-dev ffmpeg 
RUN pip install numpy==1.18.5
RUN pip install msgpack==0.5.6
RUN pip install minio requests cachetools pymongo~=3.10.1 wincertstore==0.2 opencv-contrib-python==4.2.0.34 certifi==2020.6.20 imutils Cython pillow PyYAML scipy yolov5processor==0.0.3 
ADD . /app 
WORKDIR /app 
RUN pip install edge_engine-1.0.0a0-py3-none-any.whl 
CMD ["bash","app.sh"] 