From 8403fc716b4da56521bec7a6159e3cfe7e9f5c94 Mon Sep 17 00:00:00 2001 From: Rob Ballantyne Date: Fri, 5 Apr 2024 21:54:21 +0100 Subject: [PATCH] fix tags --- .github/workflows/docker-build.yml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml index 29ee6e7..d030e5c 100644 --- a/.github/workflows/docker-build.yml +++ b/.github/workflows/docker-build.yml @@ -8,9 +8,9 @@ on: env: UBUNTU_VERSION: 22.04 BUILDX_NO_DEFAULT_ATTESTATIONS: 1 - LATEST_CUDA: "pytorch-2.2.0-py3.10-cuda-11.8.0-runtime-22.04" - LATEST_ROCM: "pytorch-2.2.0-py3.10-rocm-5.7-runtime-22.04" - LATEST_CPU: "pytorch-2.2.0-py3.10-cpu-22.04" + LATEST_CUDA: "pytorch-2.2.1-py3.10-cuda-11.8.0-runtime-22.04" + LATEST_ROCM: "pytorch-2.2.1-py3.10-rocm-5.7-runtime-22.04" + LATEST_CPU: "pytorch-2.2.1-py3.10-cpu-22.04" jobs: cpu-base: