From cde9a581e4be6b4dbe9ef2582786955c2060f605 Mon Sep 17 00:00:00 2001
From: Derek Holloway <derek@mistox.com>
Date: Sun, 17 May 2026 21:10:15 -0700
Subject: [PATCH] Update container base

---
 comfyui-cuda-dockerfile | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/comfyui-cuda-dockerfile b/comfyui-cuda-dockerfile
index dc5c263..a0245f2 100644
--- a/comfyui-cuda-dockerfile
+++ b/comfyui-cuda-dockerfile
@@ -1,12 +1,14 @@
 # Get cuda version from https://hub.docker.com/r/nvidia/cuda
-FROM nvcr.io/nvidia/pytorch:25.05-py3 AS build-deps
+FROM nvcr.io/nvidia/pytorch:26.04-py3 AS build-deps
 
 # Set ENV
 ENV USE_CUDA=1
 ENV BUILD_SOX=1
 
 # Install system packages to build torchaudio
-RUN apt update && apt install -y \
+RUN apt update && \
+    apt upgrade && \
+    apt install -y \
     libavformat-dev \
     libavcodec-dev \
     libavutil-dev \