diff --git a/comfyui-rocm-dockerfile b/comfyui-rocm-dockerfile
index 575cbba..dbc1646 100644
--- a/comfyui-rocm-dockerfile
+++ b/comfyui-rocm-dockerfile
@@ -4,11 +4,15 @@ FROM ubuntu:24.04
 # Set environment variables for 7900 XTX (gfx1100)
 ENV HSA_OVERRIDE_GFX_VERSION=11.0.0
 ENV TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL=1
+ENV LD_LIBRARY_PATH=/usr/lib/aarch64-linux-gnu:$LD_LIBRARY_PATH
 ENV PYTHONUNBUFFERED=1
 
-# Install basic Python and Graphics libs
-RUN apt update -y && apt install -y \
-    python3-pip python3-dev git libgl1-mesa-glx libglib2.0-0 \
+RUN apt update -y && apt install -y --no-install-recommends \
+    python3-pip \
+    python3-dev \
+    git \
+    libgl1 \
+    libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
 # Set working directory