From c6fd06369f91a098450962d5688df58f75272f28 Mon Sep 17 00:00:00 2001 From: Greg Hendrickson Date: Tue, 3 Feb 2026 06:45:27 +0000 Subject: [PATCH] fix: Upgrade torch via pip instead of new base image - Use known-working base image (2.4.0) - Upgrade torch to 2.6+ via pip - Previous base image was too slow to pull --- components/runpod_trainer/Dockerfile | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/components/runpod_trainer/Dockerfile b/components/runpod_trainer/Dockerfile index 7c58581..5394be6 100644 --- a/components/runpod_trainer/Dockerfile +++ b/components/runpod_trainer/Dockerfile @@ -1,12 +1,13 @@ -FROM runpod/pytorch:1.0.3-cu1290-torch260-ubuntu2204 +FROM runpod/pytorch:2.4.0-py3.11-cuda12.4.1-devel-ubuntu22.04 WORKDIR /app # Copy requirements first for better caching COPY requirements.txt /app/requirements.txt -# Install dependencies (torch 2.8 in base image satisfies CVE-2025-32434) +# Upgrade torch to 2.6+ for CVE-2025-32434 compliance, then install deps RUN pip install --upgrade pip && \ + pip install --no-cache-dir "torch>=2.6.0" torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124 && \ pip install --no-cache-dir -r requirements.txt # Copy handler and data