File size: 851 Bytes
63089c1 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | # 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"]
|