# Optional Dockerfile. On RunPod, it is usually easier to start from a PyTorch
# 2.4.0 / CUDA 12.4 template and run runpod/setup_runpod.sh.
FROM nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04

ENV DEBIAN_FRONTEND=noninteractive \
    HF_HOME=/workspace/hf_cache \
    TRANSFORMERS_CACHE=/workspace/hf_cache \
    PYTHONUNBUFFERED=1

RUN apt-get update && apt-get install -y --no-install-recommends \
    software-properties-common git curl wget ca-certificates build-essential \
    && add-apt-repository ppa:deadsnakes/ppa -y \
    && apt-get update && apt-get install -y --no-install-recommends \
    python3.11 python3.11-venv python3.11-dev \
    && rm -rf /var/lib/apt/lists/*

WORKDIR /workspace/protomorph_dinov3_runpod
COPY . /workspace/protomorph_dinov3_runpod
RUN bash runpod/setup_runpod.sh

EXPOSE 8888
CMD ["bash", "runpod/start_jupyter.sh"]