FROM ubuntu:22.04 as base

RUN apt-get update 

ENV TZ="Asia/Taipei"
RUN apt-get install -yq  \
    build-essential \
    curl \
    libncursesw5-dev \
    libssl-dev \
    libsqlite3-dev \
    libgdbm-dev \
    libc6-dev \
    libbz2-dev \
    software-properties-common \
    python3.11 python3.11-dev python3-pip

RUN apt-get install -yq  git
RUN pip3 config set global.index-url https://mirror.baidu.com/pypi/simple
RUN pip3 config set global.trusted-host mirror.baidu.com
RUN pip3 install --upgrade pip
RUN pip3 install torch==2.0.1 
RUN pip3 install torch-model-archiver==0.8.2
RUN pip3 install torchvision==0.15.2
COPY requirements.txt .

WORKDIR /docgpt
ENV PYTHONPATH=/docgpt/