diff --git a/comfyui-rocm-dockerfile b/comfyui-rocm-dockerfile
index b18718b..c5f06a1 100644
--- a/comfyui-rocm-dockerfile
+++ b/comfyui-rocm-dockerfile
@@ -8,6 +8,7 @@ RUN apt update && apt upgrade -y && apt install -y \
     git \
     curl \
     wget \
+    cmake \
     golang-go \
     ca-certificates