mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 11:56:07 +00:00
Remove 10.2 cuda support
PiperOrigin-RevId: 402707900
This commit is contained in:
parent
7a3bf61e89
commit
66a4a9ff3f
@ -8,6 +8,8 @@ Remember to align the itemized text with the first line of an item within a list
|
||||
PLEASE REMEMBER TO CHANGE THE '..main' WITH AN ACTUAL TAG in GITHUB LINK.
|
||||
-->
|
||||
|
||||
## jaxlib 0.1.73 (Unreleased)
|
||||
|
||||
## jax 0.2.22 (Unreleased)
|
||||
* [GitHub
|
||||
commits](https://github.com/google/jax/compare/jax-v0.2.21...main).
|
||||
@ -40,9 +42,10 @@ PLEASE REMEMBER TO CHANGE THE '..main' WITH AN ACTUAL TAG in GITHUB LINK.
|
||||
the `--experimental_cpp_pmap` flag (or `JAX_CPP_PMAP` environment variable).
|
||||
* `jax.numpy.unique` now supports an optional `fill_value` argument ({jax-issue}`#8121`)
|
||||
|
||||
## jaxlib 0.1.73 (Unreleased)
|
||||
|
||||
## jaxlib 0.1.72 (Oct 12, 2021)
|
||||
* Breaking changes:
|
||||
* Support for CUDA 10.2 and CUDA 10.1 has been dropped. Jaxlib now supports
|
||||
CUDA 11.1+.
|
||||
|
||||
## jax 0.2.21 (Sept 23, 2021)
|
||||
* [GitHub
|
||||
|
@ -436,7 +436,6 @@ The jaxlib version must correspond to the version of the existing CUDA
|
||||
installation you want to use:
|
||||
* For CUDA 11.1, 11.2, or 11.3, use `cuda111`. The same wheel should work for
|
||||
CUDA 11.x releases from 11.1 onwards.
|
||||
* For CUDA 10.2, use `cuda102` (will be dropped in the next release).
|
||||
* Older CUDA versions are not supported.
|
||||
|
||||
You can find your CUDA version with the command:
|
||||
|
2
setup.py
2
setup.py
@ -16,7 +16,7 @@ from setuptools import setup, find_packages
|
||||
|
||||
# The following should be updated with each new jaxlib release.
|
||||
_current_jaxlib_version = '0.1.72'
|
||||
_available_cuda_versions = ['102', '111']
|
||||
_available_cuda_versions = ['111']
|
||||
|
||||
_dct = {}
|
||||
with open('jax/version.py') as f:
|
||||
|
Loading…
x
Reference in New Issue
Block a user