commit 68e6c8da5a56b4895114ec47fc6aeccb8027de88
author Saurabh Saxena <srbs@google.com> Mon Apr 05 09:50:14 2021 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Mon Apr 05 09:59:39 2021 -0700
tree 8a60cd609efca51036b845a943cb7a28bf64e162
parent 8869fdbc2f194b9eb4e10f922d8743288bdb9810

   Update CUDA version to 11.2

PiperOrigin-RevId: 366816316
Change-Id: I46602df94f69f1802491028207445dd99d4c191a

tensorflow/tools/pip_package/setup.py

diff --git a/tensorflow/tools/pip_package/setup.py b/tensorflow/tools/pip_package/setup.py
index 17a85a5..62e66c9 100644
--- a/tensorflow/tools/pip_package/setup.py
+++ b/tensorflow/tools/pip_package/setup.py
@@ -336,7 +336,7 @@
     classifiers=sorted([
         'Development Status :: 5 - Production/Stable',
         # TODO(angerson) Add IFTTT when possible
-        'Environment :: GPU :: NVIDIA CUDA :: 11.2',
+        'Environment :: GPU :: NVIDIA CUDA :: 11.0',
         'Intended Audience :: Developers',
         'Intended Audience :: Education',
         'Intended Audience :: Science/Research',