| # 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"] | |