FROM python:3.10.12-slim-bookworm

ENV DEBIAN_FRONTEND=noninteractive

WORKDIR /melanoma-classification

RUN apt-get update && apt-get install -y \
   gcc \
   libglib2.0-0 \
   libsm6 \
   libxext6 \
   libxrender-dev \
   dos2unix \
   libgl1 \
   libglib2.0-0 \
   && rm -rf /var/lib/apt/lists/*

RUN pip install --upgrade pip

RUN pip install numpy==1.24.3

RUN pip install \
   torch==2.2.0+cpu \
   torchvision==0.17.0+cpu \
   torchaudio==2.2.0+cpu \
   -f https://download.pytorch.org/whl/cpu/torch_stable.html

RUN pip install \
   timm==0.9.16 \
   tensorboardX \
   six \
   submitit \
   fairlearn \
   kneed \
   torchsampler \
   tqdm \
   opencv-python \
   seaborn \
   onnx \
   onnxruntime \
   pandas

COPY ../ /melanoma-classification

RUN find . -type f -name "*.sh" -exec dos2unix {} \;
CMD ["/bin/bash"]