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 \