Remove 10.2 cuda support

PiperOrigin-RevId: 402707900
This commit is contained in:
Yash Katariya 2021-10-12 18:43:40 -07:00 committed by jax authors
parent 7a3bf61e89
commit 66a4a9ff3f
3 changed files with 6 additions and 4 deletions

View File

@ -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

View File

@ -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:

View File

@ -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: