From 84c22609bd6fe539b5f353f1b9c726ce17b2c059 Mon Sep 17 00:00:00 2001 From: taco Date: Fri, 12 Jun 2026 17:10:55 -0600 Subject: [PATCH] fix rocm build --- Dockerfile.rocm | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile.rocm b/Dockerfile.rocm index 187ffb5..eb919d5 100644 --- a/Dockerfile.rocm +++ b/Dockerfile.rocm @@ -1,4 +1,4 @@ -FROM rocm/pytorch:py3.11-rocm67-ubuntu2404 +FROM rocm/pytorch:rocm7.1.1_ubuntu22.04_py3.11_pytorch_release_2.10.0 WORKDIR /app