[image] Add cuda 12.8.0 in image building matrix (ray-project#51210)

Add `cu12.8.0-cudnn` in image building matrix and other relevant files &
tests

---------

Signed-off-by: kevin <[email protected]>

Author: khluu

.buildkite/_forge.rayci.yml

@@ -26,6 +26,7 @@ steps:
           - "12.3.2-cudnn9"
           - "12.4.1-cudnn"
           - "12.5.1-cudnn"
+          - "12.8.0-cudnn"
     env:
       PYTHON_VERSION: "{{matrix.python}}"
       CUDA_VERSION: "{{matrix.cuda}}"

.buildkite/build.rayci.yml

@@ -61,7 +61,7 @@ steps:
         --platform cu11.7.1-cudnn8 --platform cu11.8.0-cudnn8
         --platform cu12.1.1-cudnn8 --platform cu12.3.2-cudnn9
         --platform cu12.4.1-cudnn --platform cu12.5.1-cudnn
-        --platform cpu
+        --platform cu12.8.0-cudnn --platform cpu
         --image-type ray --upload
     depends_on:
       - manylinux

.buildkite/linux_aarch64.rayci.yml

@@ -34,6 +34,7 @@ steps:
           - "12.3.2-cudnn9"
           - "12.4.1-cudnn"
           - "12.5.1-cudnn"
+          - "12.8.0-cudnn"
     instance_type: builder-arm64
     env:
       PYTHON_VERSION: "{{matrix.python}}"
@@ -85,7 +86,7 @@ steps:
         --platform cu11.7.1-cudnn8 --platform cu11.8.0-cudnn8
         --platform cu12.1.1-cudnn8 --platform cu12.3.2-cudnn9
         --platform cu12.4.1-cudnn --platform cu12.5.1-cudnn
-        --platform cpu
+        --platform cu12.8.0-cudnn --platform cpu
         --image-type ray --architecture aarch64 --upload
     depends_on:
       - manylinux-aarch64

ci/ray_ci/docker_container.py

@@ -15,6 +15,7 @@
     "cu12.3.2-cudnn9",
     "cu12.4.1-cudnn",
     "cu12.5.1-cudnn",
+    "cu12.8.0-cudnn",
 ]
 PLATFORMS_RAY_ML = [
     "cpu",

ci/ray_ci/test_ray_docker_container.py

@@ -402,6 +402,9 @@ def test_get_platform_tag(self) -> None:
         container = RayDockerContainer(v, "cu12.5.1-cudnn", "ray")
         assert container.get_platform_tag() == "-cu125"
 
+        container = RayDockerContainer(v, "cu12.8.0-cudnn", "ray")
+        assert container.get_platform_tag() == "-cu128"
+
     def test_should_upload(self) -> None:
         v = DEFAULT_PYTHON_VERSION
         test_cases = [

docker/retag-lambda/cuda_versions.txt

@@ -1,3 +1,4 @@
+cu128
 cu125
 cu124
 cu123