Skip to content

Commit 8d0ced2

Browse files
committed
Update build-oneflow.yml
1 parent f644f8e commit 8d0ced2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.github/workflows/build-oneflow.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,11 +37,11 @@ jobs:
3737
- tag-suffix: "cuda12.6"
3838
policy: "manylinux2014"
3939
platform: "x86_64"
40-
CUDA_BASE_IMAGE: "12.6.2-cudnn-devel-rockylinux8"
40+
CUDA_BASE_IMAGE: "nvidia/cuda:12.6.2-cudnn-devel-rockylinux8"
4141
- tag-suffix: "cuda12.5"
4242
policy: "manylinux2014"
4343
platform: "x86_64"
44-
CUDA_BASE_IMAGE: "12.5.2-cudnn-devel-rockylinux8"
44+
CUDA_BASE_IMAGE: "nvidia/cuda:12.5.2-cudnn-devel-rockylinux8"
4545
- tag-suffix: "cuda12.4"
4646
policy: "manylinux2014"
4747
platform: "x86_64"

0 commit comments

Comments
 (0)