diff --git a/docker/rocm/rocm.hcl b/docker/rocm/rocm.hcl
index 0e9ae2c5c..6a52b9578 100644
--- a/docker/rocm/rocm.hcl
+++ b/docker/rocm/rocm.hcl
@@ -2,7 +2,7 @@ variable "AMDGPU" {
   default = "gfx900"
 }
 variable "ROCM" {
-  default = "7.1.0"
+  default = "7.1"
 }
 variable "HSA_OVERRIDE_GFX_VERSION" {
   default = ""