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

Fix pip dependency after cuDNN 9 upgrade

Browse Source 
PiperOrigin-RevId: 637959308
This commit is contained in:
Henning Becker 2024-05-28 11:06:06 -07:00 committed by jax authors
parent 43f51d73ce
commit 437263659c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@ -22,7 +22,7 @@ project_name = 'jax'
_current_jaxlib_version = '0.4.28'
# The following should be updated with each new jaxlib release.
_latest_jaxlib_version_on_pypi = '0.4.28'
_default_cuda12_cudnn_version = '89'
_default_cuda12_cudnn_version = '91'
_available_cuda12_cudnn_versions = [_default_cuda12_cudnn_version]
_libtpu_version = '0.1.dev20240508'