diff --git a/Dockerfile.rocm b/Dockerfile.rocm
index 187ffb5..eb919d5 100644
--- a/Dockerfile.rocm
+++ b/Dockerfile.rocm
@@ -1,4 +1,4 @@
-FROM rocm/pytorch:py3.11-rocm67-ubuntu2404
+FROM rocm/pytorch:rocm7.1.1_ubuntu22.04_py3.11_pytorch_release_2.10.0
 
 WORKDIR /app