From ed3ba5632a85f0dc3524873ccfc560d417a474af Mon Sep 17 00:00:00 2001 From: "Banikumar Maiti (Intel)" Date: Fri, 28 Oct 2022 10:33:15 -0700 Subject: [PATCH] Show warning for Windows GPU build --- configure.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/configure.py b/configure.py index bf338bdda22..6f11bb69a7d 100644 --- a/configure.py +++ b/configure.py @@ -1221,8 +1221,16 @@ 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):