diff --git a/Dockerfile.rocm b/Dockerfile.rocm
index 49db5e9..43812c4 100644
--- a/Dockerfile.rocm
+++ b/Dockerfile.rocm
@@ -1,4 +1,4 @@
-FROM rocm/dev-ubuntu-22.04:7.2
+FROM rocm/dev-ubuntu-22.04:6.4
 
 ENV DEBIAN_FRONTEND=noninteractive \
     PYTHONDONTWRITEBYTECODE=1 \
diff --git a/requirements-rocm-init.txt b/requirements-rocm-init.txt
index 8627435..ce4fabe 100644
--- a/requirements-rocm-init.txt
+++ b/requirements-rocm-init.txt
@@ -1,5 +1,5 @@
---index-url https://download.pytorch.org/whl/rocm7.2
-torch==2.11.0
-torchaudio==2.11.0
-torchvision==0.26.0
-pytorch_triton_rocm==3.3.0
+--index-url https://download.pytorch.org/whl/rocm6.4
+torch==2.9.1
+torchaudio==2.9.1
+torchvision==0.24.1
+pytorch_triton_rocm==3.5.1