mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bump minimum supported CUDNN version to 9.0

Browse Source 
This should have been changed for the 0.4.30 release; updating this value
will lead to better errors when attempting to install with an older cudnn.
This commit is contained in:
Jake VanderPlas 2024-06-27 14:56:51 -07:00
parent af8bdd1582
commit 337cefa7bb
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jax/_src/xla_bridge.py
View File

@ -366,7 +366,7 @@ def _check_cuda_versions(raise_on_first_error: bool = False,
# versions:
# https://docs.nvidia.com/deeplearning/cudnn/developer-guide/index.html#api-compat
scale_for_comparison=100,
min_supported_version=8900
min_supported_version=9000
)
_version_check("cuFFT", cuda_versions.cufft_get_version,
cuda_versions.cufft_build_version,