Skip to content

Commit 1f8a2c7

Browse files
committed
Updated GPU Support info
1 parent 7486c4d commit 1f8a2c7

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

site/en/install/source_windows.md

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -171,7 +171,6 @@ Preconfigured Bazel build configs to DISABLE default on features:
171171
--config=nogcp # Disable GCP support.
172172
--config=nonccl # Disable NVIDIA NCCL support.
173173

174-
# Configuration finished
175174
</pre>
176175
</section>
177176

@@ -212,6 +211,8 @@ bazel build --config=opt //tensorflow/tools/pip_package:build_pip_package
212211

213212
#### GPU support
214213

214+
Note: This is only for 2.10 or earlier versions, 2.11 or later versions do not support GPU on native-Windows
215+
215216
To make the TensorFlow package builder with GPU support:
216217

217218
<pre class="devsite-terminal tfo-terminal-windows devsite-click-to-copy">
@@ -308,6 +309,8 @@ For GPU support, add the CUDA and cuDNN bin directories to your `$PATH`:
308309
<code class="devsite-terminal">export PATH="/c/tools/cuda/bin:$PATH"</code>
309310
</pre>
310311

312+
Note: 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 or use TensorFlow-DirectML-Plugin
313+
311314
<a name="tested_build_configurations"></a>
312315
## Tested build configurations
313316

@@ -347,6 +350,7 @@ For GPU support, add the CUDA and cuDNN bin directories to your `$PATH`:
347350
</table>
348351

349352
### GPU
353+
Note: 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 or use TensorFlow-DirectML-Plugin
350354

351355
<table>
352356
<tr><th>Version</th><th>Python version</th><th>Compiler</th><th>Build tools</th><th>cuDNN</th><th>CUDA</th></tr>

0 commit comments

Comments
 (0)