diff --git a/docker/Dockerfile-rocm b/docker/Dockerfile-rocm
index 504660848..e93194940 100644
--- a/docker/Dockerfile-rocm
+++ b/docker/Dockerfile-rocm
@@ -37,7 +37,7 @@ COPY . /milabench/milabench/
 # build-essential: for rust
 
 RUN apt-get update &&\
-    apt-get install -y git build-essential curl &&\
+    apt-get install -y git build-essential curl python3.10 &&\
     apt-get clean &&\
     rm -rf /var/lib/apt/lists/*