experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Change compute capability < 3.5 check from error into warning. See issue #26394.

Browse Source 
PiperOrigin-RevId: 237971151
This commit is contained in:
A. Unique TensorFlower 2019-03-12 01:41:05 -07:00 committed by TensorFlower Gardener
parent 8c99eb5f8b
commit 8dc2d0eeda
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
configure.py
View File

@ -1305,11 +1305,14 @@ def set_tf_cuda_compute_capabilities(environ_cp):
all_valid = False
else:
ver = float(m.group(0))
if ver < 3.5:
print('ERROR: TensorFlow only supports CUDA compute capabilities 3.5 '
if ver < 3.0:
print('ERROR: TensorFlow only supports CUDA compute capabilities 3.0 '
'and higher. Please re-specify the list of compute '
'capabilities excluding version %s.' % ver)
all_valid = False
if ver < 3.5:
print('WARNING: XLA does not support CUDA compute capabilities '
'lower than 3.5. Disable XLA when running on older GPUs.')
if all_valid:
break