FROM nvcr.io/nvidia/tensorrt:22.07-py3

EXPOSE 8633
WORKDIR /app

RUN echo "nameserver 8.8.8.8" > /etc/resolv.conf && pip install --upgrade pip

RUN echo "nameserver 8.8.8.8" > /etc/resolv.conf && apt-get update && \
    DEBIAN_FRONTEND=noninteractive \
    apt-get -y install default-jre-headless && \
    apt-get clean && \
    rm -rf /var/lib/apt/lists/*

COPY req.txt /app/
RUN echo "nameserver 8.8.8.8" > /etc/resolv.conf && pip install -r req.txt
RUN echo "nameserver 8.8.8.8" > /etc/resolv.conf &&    pip install --upgrade pip  && pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

COPY ./main_cache.py /app/


RUN rm -r ~/.cache/pip/*
CMD ["python3", "main_cache.py"]