diff --git a/deps/ReactantExtra/.bazelrc b/deps/ReactantExtra/.bazelrc index ba56a9d61a..e1dba3f200 100644 --- a/deps/ReactantExtra/.bazelrc +++ b/deps/ReactantExtra/.bazelrc @@ -19,7 +19,7 @@ build:cuda --repo_env TF_NEED_CUDA=1 build:cuda --repo_env TF_NVCC_CLANG=1 build:cuda --repo_env TF_NCCL_USE_STUB=1 build:cuda --repo_env=HERMETIC_CUDA_VERSION="12.6.2" -build:cuda --repo_env=HERMETIC_CUDNN_VERSION="9.6.0" +build:cuda --repo_env=HERMETIC_CUDNN_VERSION="9.4.0" # "sm" means we emit only cubin, which is forward compatible within a GPU generation. # "compute" means we emit both cubin and PTX, which is larger but also forward compatible to future GPU generations. build:cuda --repo_env HERMETIC_CUDA_COMPUTE_CAPABILITIES="sm_50,sm_60,sm_70,sm_80,compute_90"