Update CUDA SDK to 12.8 and CUDNN to 9.7.0

PiperOrigin-RevId: 719198653

Author: beckerhe

.bazelrc

@@ -225,8 +225,8 @@ build:cuda --repo_env TF_NEED_CUDA=1
 build:cuda --crosstool_top=@local_config_cuda//crosstool:toolchain
 build:cuda --@local_config_cuda//:enable_cuda
 # Default CUDA and CUDNN versions.
-build:cuda --repo_env=HERMETIC_CUDA_VERSION="12.5.1"
-build:cuda --repo_env=HERMETIC_CUDNN_VERSION="9.3.0"
+build:cuda --repo_env=HERMETIC_CUDA_VERSION="12.8.0"
+build:cuda --repo_env=HERMETIC_CUDNN_VERSION="9.7.0"
 # This flag is needed to include CUDA libraries.
 build:cuda --@local_config_cuda//cuda:include_cuda_libs=true
 
@@ -256,8 +256,8 @@ build:cuda_clang --linkopt="-lm"
 
 # Set up compilation CUDA version and paths and use the CUDA Clang toolchain.
 build:cuda_clang_official --config=cuda_clang
-build:cuda_clang_official --repo_env=HERMETIC_CUDA_VERSION="12.5.1"
-build:cuda_clang_official --repo_env=HERMETIC_CUDNN_VERSION="9.3.0"
+build:cuda_clang_official --repo_env=HERMETIC_CUDA_VERSION="12.7.0"
+build:cuda_clang_official --repo_env=HERMETIC_CUDNN_VERSION="9.7.0"
 build:cuda_clang_official --action_env=CLANG_CUDA_COMPILER_PATH="/usr/lib/llvm-18/bin/clang"
 build:cuda_clang_official --crosstool_top="@local_config_cuda//crosstool:toolchain"
 

third_party/gpus/cuda/hermetic/cuda_redist_versions.bzl

@@ -70,6 +70,10 @@ CUDA_REDIST_JSON_DICT = {
         "https://developer.download.nvidia.com/compute/cuda/redist/redistrib_12.6.3.json",
         "9c598598457a6463eb92889080c16b2b9dc04150e501b8bfc1536d403ba70aaf",
     ],
+    "12.8.0": [
+        "https://developer.download.nvidia.com/compute/cuda/redist/redistrib_12.8.0.json",
+        "daa0d766b36feaa933592162c27be5fb63b68fc547ca6886c160a35d96ee8891",
+    ],
 }
 
 CUDNN_REDIST_JSON_DICT = {
@@ -121,6 +125,10 @@ CUDNN_REDIST_JSON_DICT = {
         "https://developer.download.nvidia.com/compute/cudnn/redist/redistrib_9.6.0.json",
         "6dd9a931d981fe5afc7e7ed0c422a4035b1411db4e28a39cf2429e62e3efcd3e",
     ],
+    "9.7.0": [
+        "https://developer.download.nvidia.com/compute/cudnn/redist/redistrib_9.7.0.json",
+        "e715c1d028585d228c4678c2cdc5ad9a34fde54515a1c52aa60e36021a90dd90",
+    ],
 }
 
 CUDA_12_NCCL_WHEEL_DICT = {
@@ -159,12 +167,14 @@ CUDA_NCCL_WHEELS = {
     "12.6.1": CUDA_12_NCCL_WHEEL_DICT,
     "12.6.2": CUDA_12_NCCL_WHEEL_DICT,
     "12.6.3": CUDA_12_NCCL_WHEEL_DICT,
+    "12.8.0": CUDA_12_NCCL_WHEEL_DICT,
 }
 
 REDIST_VERSIONS_TO_BUILD_TEMPLATES = {
     "nvidia_driver": {
         "repo_name": "cuda_driver",
         "version_to_template": {
+            "570": "//third_party/gpus/cuda/hermetic:cuda_driver.BUILD.tpl",
             "560": "//third_party/gpus/cuda/hermetic:cuda_driver.BUILD.tpl",
             "555": "//third_party/gpus/cuda/hermetic:cuda_driver.BUILD.tpl",
             "550": "//third_party/gpus/cuda/hermetic:cuda_driver.BUILD.tpl",