From 32191c688a63f14aefbb27ca4076ef7681078cc8 Mon Sep 17 00:00:00 2001
From: Kishor Prins <me@68x.ca>
Date: Wed, 30 Apr 2025 08:25:08 -0700
Subject: [PATCH] Use ROCm 6.4 and smaller base Docker image

---
 docker/rocm/Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker/rocm/Dockerfile b/docker/rocm/Dockerfile
index a24f336..3f4a1b9 100644
--- a/docker/rocm/Dockerfile
+++ b/docker/rocm/Dockerfile
@@ -1,4 +1,4 @@
-FROM rocm/pytorch:rocm6.3.4_ubuntu22.04_py3.10_pytorch_release_2.4.0
+FROM rocm/dev-ubuntu-24.04:6.4
 ENV DEBIAN_FRONTEND=noninteractive \
     PHONEMIZER_ESPEAK_PATH=/usr/bin \
     PHONEMIZER_ESPEAK_DATA=/usr/share/espeak-ng-data \