diff --git a/CHANGELOG.md b/CHANGELOG.md index c6592d602..9498d19f3 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -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 diff --git a/README.md b/README.md index b6709f550..7b335e9fd 100644 --- a/README.md +++ b/README.md @@ -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: diff --git a/setup.py b/setup.py index 046abe3a7..7b025d10a 100644 --- a/setup.py +++ b/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: