Update Dockerfile

Author: danielckv

Dockerfile

@@ -1,55 +1,56 @@
-# 1. PLATFORM PINNING: Essential for CUDA wheels.
-# PyTorch with CUDA does not have wheels for ARM64 (Apple Silicon).
-# We force linux/amd64 so Docker pulls the compatible x86 binary.
+# 1. PLATFORM PINNING
 FROM --platform=linux/amd64 python:3.11-slim
 
 # 2. ENVIRONMENT VARIABLES
-# Keeps Python from buffering stdout/stderr (logs appear immediately)
-# and prevents python from writing .pyc files.
 ENV PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1 \
-    PIP_NO_CACHE_DIR=1
+    PIP_NO_CACHE_DIR=1 \
+    TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6;8.9;9.0"
 
 WORKDIR /app
 
 # 3. SYSTEM DEPENDENCIES
-# Install basic build tools and libraries often required by vision/audio packages
 RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential \
     git \
     libgl1 \
     libglib2.0-0 \
+    # wget/curl often useful in notebooks
+    wget \
+    curl \
     && rm -rf /var/lib/apt/lists/*
 
 # 4. INSTALL PYTORCH (Heavy Layer)
-# We do this BEFORE copying requirements.txt or app code.
-# This ensures Docker caches this heavy layer (2GB+) and doesn't re-download
-# it unless you specifically change the Torch version.
 RUN pip install --upgrade pip setuptools wheel && \
     pip install \
     torch==2.7.0 \
     torchvision==0.22.0 \
     torchaudio==2.7.0 \
     --index-url https://download.pytorch.org/whl/cu126
 
-# 5. INSTALL OTHER REQUIREMENTS
-COPY requirements.txt .
-RUN pip install -r requirements.txt
+# 5. CLONE SAM 3 & INSTALL JUPYTER
+# We install sam3 AND jupyterlab here
+RUN git clone https://github.com/facebookresearch/sam3.git && \
+    cd sam3 && \
+    pip install -e . && \
+    pip install jupyterlab matplotlib ipywidgets
 
-# 6. SECURITY: CREATE NON-ROOT USER
-# Running as root is a security risk. Create a user 'appuser'.
+# 6. SECURITY: NON-ROOT USER
 RUN addgroup --system --gid 1001 appgroup && \
     adduser --system --uid 1001 --gid 1001 appuser
 
-# 7. COPY APP CODE
-COPY . .
-
-# Change ownership of the app directory to the non-root user
+# Grant ownership so Jupyter can write notebooks to /app
 RUN chown -R appuser:appgroup /app
 
 # Switch to non-root user
 USER appuser
 
+# 7. CONFIGURATION
+# Expose the standard Jupyter port
+EXPOSE 8888
+
 # 8. ENTRYPOINT
-# Update 'main.py' to your actual entry script
-CMD ["python", "main.py"]
+# --ip=0.0.0.0: Allows connections from outside the container
+# --no-browser: Prevents it from trying to open a browser inside the container
+# --NotebookApp.token='': (Optional) Disables password for easier local dev
+CMD ["jupyter", "lab", "--ip=0.0.0.0", "--port=8888", "--no-browser", "--NotebookApp.token=''"]