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

Merge pull request #58356 from Intel-tensorflow:bani/win_tf2.11_warn_onlycpu

Browse Source 
PiperOrigin-RevId: 484832624
This commit is contained in:
TensorFlower Gardener
2022-10-30 00:45:48 -07:00
parent 35e982d715 ed3ba5632a
commit 465d3aaf56
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
configure.py
View File

@@ -1221,8 +1221,15 @@ def main():
if (environ_cp.get('TF_NEED_ROCM') == '1' and environ_cp.get('HIP_PLATFORM')):
write_action_env_to_bazelrc('HIP_PLATFORM', environ_cp.get('HIP_PLATFORM'))
environ_cp['TF_NEED_CUDA'] = str(
int(get_var(environ_cp, 'TF_NEED_CUDA', 'CUDA', False)))
if is_windows():
print('\nWARNING: Cannot build with CUDA support on Windows.\n'
'Starting in TF 2.11, CUDA build is not supported for Windows. '
'For using TensorFlow GPU on Windows, you will need to build/install '
'TensorFlow in WSL2.\n')
environ_cp['TF_NEED_CUDA'] = '0'
else:
environ_cp['TF_NEED_CUDA'] = str(
int(get_var(environ_cp, 'TF_NEED_CUDA', 'CUDA', False)))
if (environ_cp.get('TF_NEED_CUDA') == '1' and
'TF_CUDA_CONFIG_REPO' not in environ_cp):