ghndrx/kubeflow-pipelines
1
0
Fork 0
mirror of https://github.com/ghndrx/kubeflow-pipelines.git synced 2026-02-10 06:45:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: Upgrade torch via pip instead of new base image

Browse Source 
- Use known-working base image (2.4.0)
- Upgrade torch to 2.6+ via pip
- Previous base image was too slow to pull
This commit is contained in:
Greg Hendrickson
2026-02-03 06:45:27 +00:00
parent 9b66b5fd14
commit c6fd06369f
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
components/runpod_trainer/Dockerfile
View File

@@ -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 WORKDIR /app
# Copy requirements first for better caching # Copy requirements first for better caching
COPY requirements.txt /app/requirements.txt 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 && \ 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 pip install --no-cache-dir -r requirements.txt
# Copy handler and data # Copy handler and data