From c6ce11b421657d054ce1686c9a0b7d07e7db5790 Mon Sep 17 00:00:00 2001 From: Benjamin Berman Date: Tue, 30 Jul 2024 11:42:14 -0700 Subject: [PATCH] Switch to supported ROCm image --- .github/workflows/test.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index 33f09c0da..9fd775111 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -53,7 +53,7 @@ jobs: - labels: [self-hosted, Linux, X64, cpu] container: "ubuntu:22.04" - labels: [self-hosted, Linux, X64, rocm-7600-8gb] - container: "rocm/pytorch:rocm6.1_ubuntu22.04_py3.10_pytorch_2.4" + container: "rocm/pytorch:rocm6.1.3_ubuntu22.04_py3.10_pytorch_release-2.1.2" - labels: [self-hosted, Linux, X64, cuda-3060-12gb] container: "nvcr.io/nvidia/pytorch:24.03-py3" steps: