diff --git a/Dockerfile.aiter b/Dockerfile.aiter
index b61c1e41a5..dab3f9588d 100644
--- a/Dockerfile.aiter
+++ b/Dockerfile.aiter
@@ -1,9 +1,8 @@
-ARG BASE_DOCKER="rocm/composable_kernel-private:ck_aiter_base"
+ARG BASE_DOCKER="rocm/pytorch:latest"
 FROM $BASE_DOCKER
 ARG AITER_BRANCH="main"
 ARG CK_AITER_BRANCH="develop"
-RUN groupadd irc && \
-    pip install pandas zmq einops && \
+RUN pip install pandas zmq einops ninja && \
     pip install numpy==1.26.2 && \
     sudo mkdir /home/jenkins && \
     sudo mkdir /home/jenkins/workspace && \
@@ -14,6 +13,8 @@ RUN groupadd irc && \
     rm -rf 3rdparty/composable_kernel/ && \
     git clone -b "$CK_AITER_BRANCH" https://github.com/ROCm/composable_kernel.git 3rdparty/composable_kernel/ && \
     python3 setup.py develop && \
-    chown -R jenkins:jenkins /home/jenkins/workspace && \
-    chmod -R a+rwx /home/jenkins/workspace && \
+    groupadd -g 1001 jenkins && \
+    useradd -u 1001 -g 1001 -m -s /bin/bash jenkins && \
+    chown -R jenkins:jenkins /home/jenkins && \
+    chmod -R a+rwx /home/jenkins && \
     sudo usermod -aG irc jenkins