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

Add cudnn 86 (for cuda 11.8) so that I can release cuda 11.8 nightlies.

Browse Source 
PiperOrigin-RevId: 493086060
This commit is contained in:
Yash Katariya 2022-12-05 12:49:33 -08:00 committed by jax authors
parent dad21d3d79
commit 25d1a0b4c6
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
setup.py
View File

@ -24,8 +24,8 @@ _current_jaxlib_version = '0.3.25'
_latest_jaxlib_version_on_pypi = '0.3.25'
_available_cuda_versions = ['11']
_default_cuda_version = '11'
_available_cudnn_versions = ['82', '805']
_default_cudnn_version = '82'
_available_cudnn_versions = ['82', '86']
_default_cudnn_version = '86'
_libtpu_version = '0.1.dev20221109'
_dct = {}