diff --git a/comfyui-rocm-dockerfile b/comfyui-rocm-dockerfile
index a125cea..d786e55 100644
--- a/comfyui-rocm-dockerfile
+++ b/comfyui-rocm-dockerfile
@@ -6,9 +6,7 @@ ENV PIP_BREAK_SYSTEM_PACKAGES=1
 ENV PYTHONUNBUFFERED=1
 
 RUN apt update -y && apt install -y --no-install-recommends \
-    rocm-dev \
-    rocm-libs \
-    rocminfo \
+    rocm \
     python3-pip \
     python3-dev \
     python3-torch-rocm \