Dockerfile

# docker build -t registry.webis.de/code-research/authorship/pan24-generative-authorship-detection .
FROM nvcr.io/nvidia/cuda:12.3.1-devel-ubuntu22.04

RUN set -x \
    && apt update \
    && apt install -y git python3 python3-pip \
    && python3 -m pip install --no-cache poetry poetry-plugin-export \
    && rm -rf /var/lib/apt/lists/*

RUN set -x && mkdir /opt/pan24-generative-authorship-detection
WORKDIR /opt/pan24-generative-authorship-detection

# Install dependencies before copying actual source files
COPY pyproject.toml /opt/pan24-generative-authorship-detection
COPY poetry.lock /opt/pan24-generative-authorship-detection
RUN set -x \
      && poetry export > requirements.txt \
      && pip --no-cache install -r requirements.txt \
      && rm requirements.txt

#RUN set -x \
#      && poetry export --without-hashes | sed 's/==.*//' | sed '/transformers\|nvidia\|torch/d' > requirements.txt \
#      && pip --no-cache install -r requirements.txt \
#      && rm requirements.txt

# Install flash-attn separately, as it cannot be installed with build isolation and thus Poetry right now
RUN set -x \
      && pip --no-cache install packaging \
      && MAX_JOBS=8 pip --no-cache install --no-build-isolation flash-attn

COPY . /opt/pan24-generative-authorship-detection

RUN set -x && pip --no-cache install --no-deps .