From 8fb5d9bda569ba1a7c21463eaeb97478108db4e4 Mon Sep 17 00:00:00 2001 From: Shenghang Tsai Date: Thu, 14 Nov 2024 13:52:45 +0800 Subject: [PATCH] Update build-oneflow.yml --- .github/workflows/build-oneflow.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/build-oneflow.yml b/.github/workflows/build-oneflow.yml index 97b95785..dc404a41 100644 --- a/.github/workflows/build-oneflow.yml +++ b/.github/workflows/build-oneflow.yml @@ -41,7 +41,7 @@ jobs: - tag-suffix: "cuda12.5" policy: "manylinux2014" platform: "x86_64" - CUDA_BASE_IMAGE: "nvidia/cuda:12.5.2-cudnn-devel-rockylinux8" + CUDA_BASE_IMAGE: "nvidia/cuda:12.5.1-cudnn-devel-rockylinux8" - tag-suffix: "cuda12.4" policy: "manylinux2014" platform: "x86_64"