Create Dockerfile for CUDA and SAM3 setup

This Dockerfile sets up a development environment with CUDA 12.6, Ubuntu 24.04, and Python 3.12, including necessary system dependencies, PyTorch installation, and SAM3 cloning.

Author: danielckv

Dockerfile

@@ -0,0 +1,64 @@
+# Base image with CUDA 12.6 and Ubuntu 24.04 (includes Python 3.12)
+FROM nvidia/cuda:12.6.0-devel-ubuntu24.04
+
+# Set environment variables to non-interactive (prevents prompts during build)
+ENV DEBIAN_FRONTEND=noninteractive
+ENV PYTHONUNBUFFERED=1
+ENV PYTHONDONTWRITEBYTECODE=1
+
+# Set the working directory to /workspace (Standard for Runpod)
+WORKDIR /workspace
+
+# 1. Install System Dependencies
+# 'ffmpeg' is often required for video processing tasks in SAM3
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    git \
+    wget \
+    curl \
+    vim \
+    ffmpeg \
+    python3-pip \
+    python3-dev \
+    python3-venv \
+    && apt-get clean && rm -rf /var/lib/apt/lists/*
+
+# 2. Setup Python 3.12 as default 'python'
+# Ubuntu 24.04 ships with Python 3.12. We ensure 'python' points to it.
+RUN ln -s /usr/bin/python3 /usr/bin/python
+
+# 3. Install PyTorch 2.7+ with CUDA 12.6 support
+# Using the specific index URL provided in the SAM3 README
+RUN pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir \
+    torch==2.7.0 \
+    torchvision \
+    torchaudio \
+    --index-url https://download.pytorch.org/whl/cu126
+
+# 4. Clone and Install SAM3
+# Cloning into /workspace/sam3
+RUN git clone https://github.com/facebookresearch/sam3.git && \
+    cd sam3 && \
+    # Install the package in editable mode
+    pip install -e . && \
+    # Install optional dependencies for notebooks and dev as recommended
+    pip install -e ".[notebooks,dev,train]"
+
+# 5. (Optional) Install Jupyter for Runpod interactive mode
+RUN pip install jupyterlab
+
+# 6. Setup Entrypoint
+# Create a script to help with Hugging Face login (required for checkpoints)
+RUN echo '#!/bin/bash\n\
+echo "----------------------------------------------------------------"\n\
+echo "SAM3 Setup Complete."\n\
+echo "NOTE: You must log in to Hugging Face to download checkpoints."\n\
+echo "Run: huggingface-cli login"\n\
+echo "Then follow the instructions in the README to download weights."\n\
+echo "----------------------------------------------------------------"\n\
+exec "$@"' > /usr/local/bin/entrypoint.sh && chmod +x /usr/local/bin/entrypoint.sh
+
+ENTRYPOINT ["/usr/local/bin/entrypoint.sh"]
+
+# Default command: Start Jupyter Lab (common for Runpod) or sleep infinity
+CMD ["jupyter", "lab", "--ip=0.0.0.0", "--port=8888", "--allow-root", "--no-browser"]