OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2026-01-15 12:15:41 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Update cuda version in configure.py.

Browse Source 
This is not actually used but to prevent confusion let's bring it up to date

PiperOrigin-RevId: 427769347
Change-Id: I1ebc7797a337b98135ee038bb8c5840c8c7c7b3c
This commit is contained in:
Mihai Maruseac
2022-02-10 09:28:39 -08:00
committed by TensorFlower Gardener
parent 1167bb4e41
commit 7131c47acb
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
configure.py
View File

@@ -30,8 +30,8 @@ except ImportError:
from distutils.spawn import find_executable as which
# pylint: enable=g-import-not-at-top
_DEFAULT_CUDA_VERSION = '10'
_DEFAULT_CUDNN_VERSION = '7'
_DEFAULT_CUDA_VERSION = '11'
_DEFAULT_CUDNN_VERSION = '2'
_DEFAULT_TENSORRT_VERSION = '6'
_DEFAULT_CUDA_COMPUTE_CAPABILITIES = '3.5,7.0'