def set_hermetic_cudnn_version(environ_cp):
  """Set HERMETIC_CUDNN_VERSION."""
  ask_cudnn_version = (
      'Please specify the hermetic cuDNN version you want to use '
      'or leave empty to use the default version. '
  )
  hermetic_cudnn_version = get_from_env_or_user_or_default(
      environ_cp, 'HERMETIC_CUDNN_VERSION', ask_cudnn_version, None
  )
  if hermetic_cudnn_version:

    --experimental_cc_shared_library \
    --@local_config_cuda//:enable_cuda \
    --@local_config_cuda//cuda:include_cuda_libs=false \
    --repo_env=HERMETIC_CUDA_VERSION="12.3.2" \
    --repo_env=HERMETIC_CUDNN_VERSION="8.9.7.29" \
    "somepath(//tensorflow/tools/pip_package:wheel, " \
    "@local_config_cuda//cuda:cudart + "\
    "@local_config_cuda//cuda:cudart + "\
    "@local_config_cuda//cuda:cuda_driver + "\

build:mkl_aarch64 --config=mkl_aarch64_threadpool

# Default CUDA, CUDNN and NVSHMEM versions.
build:cuda_version --repo_env=HERMETIC_CUDA_VERSION="12.5.1"
build:cuda_version --repo_env=HERMETIC_CUDNN_VERSION="9.3.0"
build:cuda_version --repo_env=HERMETIC_NVSHMEM_VERSION="3.2.5"

# CUDA: This config refers to building CUDA op kernels with nvcc.
build:cuda --repo_env TF_NEED_CUDA=1