FROM docker.io/nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04
ENV TZ=Asia/Tokyo
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
RUN apt-get update && \
    apt-get install -y \
        git \
        make build-essential libssl-dev zlib1g-dev \
        libbz2-dev libreadline-dev libsqlite3-dev wget curl llvm \
        libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev git-lfs \
        python3.10-dev python3-pip && \
    rm -rf /var/lib/apt/lists/* && \
    git lfs install

RUN useradd -m -u 1000 user
USER user

COPY --link --chown=1000 ./requirements.txt /home/user/app/requirements.txt
WORKDIR /home/user/app

RUN pip install -U setuptools wheel
RUN pip install torch==2.5.1+cu124 --index-url https://download.pytorch.org/whl/cu124
RUN pip install -r requirements.txt

COPY --link --chown=1000 ./model_pull.py /home/user/app/model_pull.py
RUN python3.10 model_pull.py 

COPY --link --chown=1000 ./app.py /home/user/app/app.py

EXPOSE 7860
ENTRYPOINT ["python3.10", "app.py"]