From 66a4a9ff3f7e3a71e75861becf1b245bea35b7d5 Mon Sep 17 00:00:00 2001 From: Yash Katariya Date: Tue, 12 Oct 2021 18:43:40 -0700 Subject: [PATCH] Remove 10.2 cuda support PiperOrigin-RevId: 402707900 --- CHANGELOG.md | 7 +++++-- README.md | 1 - setup.py | 2 +- 3 files changed, 6 insertions(+), 4 deletions(-) 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: