From 986cf045e79c268ebc049e8922fb995d5993c261 Mon Sep 17 00:00:00 2001 From: Kishor Prins Date: Sat, 21 Jun 2025 10:08:52 -0700 Subject: [PATCH] Update to ROCm 6.4.1 --- docker/rocm/Dockerfile | 2 +- pyproject.toml | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/docker/rocm/Dockerfile b/docker/rocm/Dockerfile index 3f4a1b9..d139f06 100644 --- a/docker/rocm/Dockerfile +++ b/docker/rocm/Dockerfile @@ -1,4 +1,4 @@ -FROM rocm/dev-ubuntu-24.04:6.4 +FROM rocm/dev-ubuntu-24.04:6.4.1 ENV DEBIAN_FRONTEND=noninteractive \ PHONEMIZER_ESPEAK_PATH=/usr/bin \ PHONEMIZER_ESPEAK_DATA=/usr/share/espeak-ng-data \ diff --git a/pyproject.toml b/pyproject.toml index ce2f01b..8acdb43 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -47,10 +47,10 @@ gpu = [ "torch==2.6.0+cu124", ] rocm = [ - "torch @ https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4/torch-2.6.0%2Brocm6.4.0.git2fb0ac2b-cp310-cp310-linux_x86_64.whl", - "pytorch-triton-rocm @ https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4/pytorch_triton_rocm-3.2.0%2Brocm6.4.0.git6da9e660-cp310-cp310-linux_x86_64.whl", - "torchvision @ https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4/torchvision-0.21.0%2Brocm6.4.0.git4040d51f-cp310-cp310-linux_x86_64.whl", - "torchaudio @ https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4/torchaudio-2.6.0%2Brocm6.4.0.gitd8831425-cp310-cp310-linux_x86_64.whl", + "torch @ https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4.1/torch-2.6.0%2Brocm6.4.1.git1ded221d-cp310-cp310-linux_x86_64.whl", + "pytorch-triton-rocm @ https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4.1/pytorch_triton_rocm-3.2.0%2Brocm6.4.1.git6da9e660-cp310-cp310-linux_x86_64.whl", + "torchvision @ https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4.1/torchvision-0.21.0%2Brocm6.4.1.git4040d51f-cp310-cp310-linux_x86_64.whl", + "torchaudio @ https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4.1/torchaudio-2.6.0%2Brocm6.4.1.gitd8831425-cp310-cp310-linux_x86_64.whl", ] cpu = [ "torch==2.6.0",