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

Update changelog.

Browse Source 
Bump the minimum CUDA 12 pip package versions to the current releases.
This commit is contained in:
Peter Hawkins 2023-09-26 17:59:49 -04:00
parent 385cfc86e6
commit a2e1f1f24e
2 changed files with 17 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
CHANGELOG.md
View File

@ -19,9 +19,19 @@ Remember to align the itemized text with the first line of an item within a list
* {func}`jax.scipy.stats.mode` now returns a 0 count if the mode is taken * {func}`jax.scipy.stats.mode` now returns a 0 count if the mode is taken
across a size-0 axis, matching the behavior of `scipy.stats.mode` in SciPy across a size-0 axis, matching the behavior of `scipy.stats.mode` in SciPy
1.11. 1.11.
* Removed the "No GPU/TPU" found warning. Instead warn if, on Linux, an
NVIDIA GPU or a Google TPU are found but not used and `--jax_platforms` was
not specified.
# jaxlib 0.4.17 # jaxlib 0.4.17
* Changes:
* Python 3.12 wheels were added in this release.
* The CUDA 12 wheels now require CUDA 12.2.
* Bug fixes:
* Fixed log spam from ABSL when the JAX CPU backend was initialized.
## jax 0.4.16 (Sept 18, 2023) ## jax 0.4.16 (Sept 18, 2023)
* Changes * Changes

14
setup.py
View File

@ -123,14 +123,14 @@ setup(
'cuda12_pip': [ 'cuda12_pip': [
f"jaxlib=={_current_jaxlib_version}+cuda12.cudnn{_default_cuda12_cudnn_version}", f"jaxlib=={_current_jaxlib_version}+cuda12.cudnn{_default_cuda12_cudnn_version}",
"nvidia-cublas-cu12", "nvidia-cublas-cu12>=12.2.5.6",
"nvidia-cuda-cupti-cu12", "nvidia-cuda-cupti-cu12>=12.2.142",
"nvidia-cuda-nvcc-cu12", "nvidia-cuda-nvcc-cu12>=12.2.140",
"nvidia-cuda-runtime-cu12", "nvidia-cuda-runtime-cu12>=12.2.140",
"nvidia-cudnn-cu12>=8.9", "nvidia-cudnn-cu12>=8.9",
"nvidia-cufft-cu12", "nvidia-cufft-cu12>=11.0.8.103",
"nvidia-cusolver-cu12", "nvidia-cusolver-cu12>=11.5.2",
"nvidia-cusparse-cu12", "nvidia-cusparse-cu12>=12.1.2.141",
], ],
# Target that does not depend on the CUDA pip wheels, for those who want # Target that does not depend on the CUDA pip wheels, for those who want