Skip to content

Commit 25abfe4

Browse files
authored
fix: try building with cudnn 9.4 (#436)
1 parent 241fd14 commit 25abfe4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

deps/ReactantExtra/.bazelrc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ build:cuda --repo_env TF_NEED_CUDA=1
1919
build:cuda --repo_env TF_NVCC_CLANG=1
2020
build:cuda --repo_env TF_NCCL_USE_STUB=1
2121
build:cuda --repo_env=HERMETIC_CUDA_VERSION="12.6.2"
22-
build:cuda --repo_env=HERMETIC_CUDNN_VERSION="9.6.0"
22+
build:cuda --repo_env=HERMETIC_CUDNN_VERSION="9.4.0"
2323
# "sm" means we emit only cubin, which is forward compatible within a GPU generation.
2424
# "compute" means we emit both cubin and PTX, which is larger but also forward compatible to future GPU generations.
2525
build:cuda --repo_env HERMETIC_CUDA_COMPUTE_CAPABILITIES="sm_50,sm_60,sm_70,sm_80,compute_90"

0 commit comments

Comments
 (0)