FROM pytorch/pytorch:2.3.1-cuda12.1-cudnn8-devel

RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub \

&& apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub

RUN apt-get update && apt-get install -y python3 python3-pip libgl1 libglib2.0-0 git \

&& apt-get clean \

&& rm -rf /var/lib/apt/lists/*

RUN apt-get -y update && apt-get install -y gcc python3 python3-pip libgl1-mesa-glx libglib2.0-0 git libblas-dev liblapack-dev gfortran

ENV PROJECTDIR=/opt/app

WORKDIR $PROJECTDIR

COPY ./cosas/ $PROJECTDIR/cosas

COPY ./inference.py $PROJECTDIR

COPY ./requirements.txt $PROJECTDIR

COPY model.pth $PROJECTDIR

COPY setup.py $PROJECTDIR

COPY ./install_spams.sh $PROJECTDIR

COPY ./target_image.png $PROJECTDIR

RUN bash ./install_spams.sh

RUN python3 -m pip install --upgrade pip

RUN python3 -m pip install -r requirements.txt

RUN python3 -m pip install histomicstk --find-links https://girder.github.io/large_image_wheels

RUN python3 -m pip install SimpleITK

RUN python3 -m pip install .

RUN groupadd -r user && useradd -m --no-log-init -r -g user user

USER user

LABEL team="SMF"

LABEL maintainer1="SMF_hoheon <hoheon0509@mf.seegene.com>"

LABEL maintainer2="SMF_wonchan <wonchanjeong53@gmail.com>"

ENTRYPOINT ["python3", "inference.py"]