File size: 871 Bytes
dae5c90 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 | 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"] |