Dockerfile_gpu_Thien

FROM python:3.7.17-slim-bullseye

WORKDIR /app


RUN apt-get update && apt-get install build-essential cmake git -y

#----------------JRE (for vncorenlp)--------------
RUN apt-get update && \
    DEBIAN_FRONTEND=noninteractive \
    apt-get -y install default-jre-headless && \
    apt-get clean && \
    rm -rf /var/lib/apt/lists/*

COPY ./model /app/model
COPY ./tensorRT /app/tensorRT

RUN pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
RUN mkdir log
RUN mkdir log_run

COPY req.txt /app/
RUN pip install -r req.txt

COPY ./entity /app/entity
COPY ./function /app/function
COPY ./vncorenlp /app/vncorenlp
COPY ./main_cache_Thien.py /app/
COPY ./service_cache_Thien.py /app/
COPY ./summary.py /app/

RUN rm -r ~/.cache/pip/*

CMD ["python", "main_cache_Thien.py"]