sync from GitHub (28464ab)

image/universal/Dockerfile

@@ -1,20 +1,23 @@
-FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-runtime
+FROM nvidia/cuda:12.1.1-cudnn8-runtime-ubuntu22.04
+
+ENV DEBIAN_FRONTEND=noninteractive
+ENV PYTHONUNBUFFERED=1
 
 WORKDIR /app
 
-# Install git and wget
-RUN apt-get update && \
-    apt-get install -y git wget && \
-    apt-get clean && \
-    rm -rf /var/lib/apt/lists/*
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    python3 python3-pip git wget \
+    && rm -rf /var/lib/apt/lists/*
 
-# Copy requirements file
-COPY requirements.txt .
+RUN ln -sf /usr/bin/python3 /usr/bin/python
 
-# Install dependencies
-# Install dependencies
-RUN pip install uv && \
-    uv pip install --system --no-cache-dir -r requirements.txt
+# Install PyTorch with CUDA 12.1 (upgraded from 1.11/CUDA 11.3 for Blackwell support)
+RUN pip install --no-cache-dir \
+    torch==2.5.1 torchvision==0.20.1 \
+    --index-url https://download.pytorch.org/whl/cu121
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
 
 # Copy application code
 COPY app.py .