ci: Update CUDA 12.6 builds to CUDA 12.6.2

coreweave · Nov 4, 2024 · fcba3c3 · fcba3c3
1 parent 06c7fa5
commit fcba3c3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.github/workflows/ubuntu-20.yml b/.github/workflows/ubuntu-20.yml
@@ -69,8 +69,8 @@ jobs:
       folder: .
       dockerfile: Dockerfile.ubuntu20
       base-image: nvidia/cuda
-      base-tag: 12.6.1-cudnn-devel-ubuntu20.04
-      cuda-version: "12.6.1"
+      base-tag: 12.6.2-cudnn-devel-ubuntu20.04
+      cuda-version: "12.6.2"
       nccl-version: 2.23.4-1
       cuda-samples-version: "12.5"
       hpcx-distribution: "hpcx-v2.20-gcc-mlnx_ofed-ubuntu20.04-cuda12"
diff --git a/.github/workflows/ubuntu-22.yml b/.github/workflows/ubuntu-22.yml
@@ -69,8 +69,8 @@ jobs:
       folder: .
       dockerfile: Dockerfile.ubuntu22
       base-image: nvidia/cuda
-      base-tag: 12.6.1-cudnn-devel-ubuntu22.04
-      cuda-version: "12.6.1"
+      base-tag: 12.6.2-cudnn-devel-ubuntu22.04
+      cuda-version: "12.6.2"
       nccl-version: 2.23.4-1
       cuda-samples-version: "12.5"
       hpcx-distribution: "hpcx-v2.20-gcc-mlnx_ofed-ubuntu22.04-cuda12"