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
build:cuda --crosstool_top=@local_config_cuda//crosstool:toolchain
build:cuda --@local_config_cuda//:enable_cuda
build:cuda --config=cuda_version
# This flag is needed to include CUDA libraries.