Dockerfile

FROM python:3.9-slim
#FROM tensorflow/tensorflow:latest-gpu

# This tells girder_worker to enable gpu if possible
#LABEL com.nvidia.volumes.needed=nvidia_driver

RUN apt-get update && \
    apt-get install -y --no-install-recommends \
    git \
    && \
    apt-get clean && rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/* /var/cache/*

COPY . /opt/scw
WORKDIR /opt/scw
RUN --mount=type=cache,target=/root/.cache/pip python -m pip install histomicstk --find-links https://girder.github.io/large_image_wheels
RUN --mount=type=cache,target=/root/.cache/pip python -m pip install -e . --find-links https://girder.github.io/large_image_wheels

# Use a newer histomicstk
# Not needed if we install histomicstk from pypi
# RUN apt-get update && apt-get install -y git build-essential && \
#     git clone --depth=1 --single-branch -b master https://github.com/DigitalSlideArchive/HistomicsTK.git && \
#     cd HistomicsTK && \
#     pip install .

WORKDIR /opt/scw/dsa_run_custom_ai_models

RUN python -m slicer_cli_web.cli_list_entrypoint --list_cli
RUN python -m slicer_cli_web.cli_list_entrypoint runCustomAIModels --help

# This makes the results show up in a more timely manner
ENV PYTHONUNBUFFERED=TRUE

ENTRYPOINT ["/bin/bash", "docker-entrypoint.sh"]