diff --git a/comfyui-rocm-dockerfile b/comfyui-rocm-dockerfile
index f3311a8..7893c1e 100644
--- a/comfyui-rocm-dockerfile
+++ b/comfyui-rocm-dockerfile
@@ -1,20 +1,28 @@
 # Stage 1: Build Ollama with Vulkan support
-FROM ubuntu:24.04 AS builder
+FROM ubuntu:26.04 AS builder
 
 ENV DEBIAN_FRONTEND=noninteractive
 
-RUN apt-get update && apt-get install -y \
+RUN apt update && apt upgrade -y && apt install -y \
     build-essential \
-    cmake \
     git \
     curl \
+    wget \
     golang-go \
     libvulkan-dev \
     vulkan-tools \
     glslang-tools \
     ninja-build \
+    spirv-headers \
+    spirv-tools \
+    ca-certificates \
     && rm -rf /var/lib/apt/lists/*
 
+# Directly install glslc via Debian arm64 package
+RUN wget http://ftp.debian.org/debian/pool/main/s/shaderc/glslc_2023.2-1_arm64.deb \
+    && apt-get update && apt-get install -y ./glslc_2023.2-1_arm64.deb \
+    && rm glslc_2023.2-1_arm64.deb && rm -rf /var/lib/apt/lists/*
+
 WORKDIR /app
 RUN git clone --recursive https://github.com/ollama/ollama.git .