bump torch and migraphx versions (#266)

shivadbhavsar · web-flow · commit efd3b0c10e3a · 2025-10-07T12:30:55.000-07:00
diff --git a/Dockerfile b/Dockerfile
@@ -1,5 +1,4 @@
-FROM rocm/pytorch:rocm6.4_ubuntu22.04_py3.10_pytorch_release_2.6.0
-
+FROM rocm/pytorch:rocm6.4.4_ubuntu24.04_py3.12_pytorch_release_2.7.1
 ARG ROCM_PATH=/opt/rocm
 ARG MIGRAPHX_BRANCH=master 
 ARG GPU_ARCH="gfx900;gfx906;gfx908;gfx90a;gfx1030;gfx1100;gfx1101;gfx1102;gfx940;gfx941;gfx942;gfx950"
diff --git a/ci/base.Dockerfile b/ci/base.Dockerfile
@@ -1,7 +1,7 @@
-FROM rocm/pytorch:rocm6.4_ubuntu22.04_py3.10_pytorch_release_2.6.0
+FROM rocm/pytorch:rocm6.4.4_ubuntu24.04_py3.12_pytorch_release_2.7.1
 
 ARG ROCM_PATH=/opt/rocm
-ARG MIGRAPHX_BRANCH="rocm-6.4.0" 
+ARG MIGRAPHX_BRANCH="rocm-6.4.4" 
 ARG GPU_ARCH="gfx900;gfx906;gfx908;gfx90a;gfx1030;gfx1100;gfx1101;gfx1102;gfx940;gfx941;gfx942"
 
 # Install Dependencies: MIGraphX
diff --git a/docker/dev.Dockerfile b/docker/dev.Dockerfile
@@ -1,4 +1,4 @@
-FROM rocm/pytorch:rocm6.4_ubuntu22.04_py3.10_pytorch_release_2.6.0
+FROM rocm/pytorch:rocm6.4.4_ubuntu24.04_py3.12_pytorch_release_2.7.1
 
 ARG ROCM_PATH=/opt/rocm
 ARG MIGRAPHX_BRANCH=develop 

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-FROM rocm/pytorch:rocm6.4_ubuntu22.04_py3.10_pytorch_release_2.6.0`
	`1`	`+FROM rocm/pytorch:rocm6.4.4_ubuntu24.04_py3.12_pytorch_release_2.7.1`
`2`	`2`
`3`	`3`	`ARG ROCM_PATH=/opt/rocm`
`4`	`4`	`ARG MIGRAPHX_BRANCH=develop`