Update Dockerfile

Author: danielckv

Dockerfile

@@ -1,64 +1,58 @@
 # 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)
+# Set environment variables
 ENV DEBIAN_FRONTEND=noninteractive
 ENV PYTHONUNBUFFERED=1
 ENV PYTHONDONTWRITEBYTECODE=1
+ENV TORCH_CUDA_ARCH_LIST="8.0;8.6;8.9;9.0"
 
-# Set the working directory to /workspace (Standard for Runpod)
+# Set the working directory
 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 \
+    libgl1-mesa-glx \
+    libglib2.0-0 \
     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.
+# 2. Setup Python 3.12 as default
 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 && \
+# 3. Install PyTorch 2.7.0 with PINNED dependencies
+# We pin torchvision==0.22.0 and torchaudio==2.7.0 to prevent conflicts
+RUN pip install --no-cache-dir --upgrade pip setuptools wheel && \
     pip install --no-cache-dir \
     torch==2.7.0 \
-    torchvision \
-    torchaudio \
+    torchvision==0.22.0 \
+    torchaudio==2.7.0 \
     --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
+# 5. Install JupyterLab
 RUN pip install jupyterlab
 
-# 6. Setup Entrypoint
-# Create a script to help with Hugging Face login (required for checkpoints)
+# 6. Setup Entrypoint for Hugging Face Login
 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 "SAM3 Environment Ready."\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"]