Skip to content

Commit 085f686

Browse files
penpornkcopybara-github
authored andcommitted
Add Intel® Extension for TensorFlow to the gpu_plugins.md page. Add TF version requirements for all plug-ins.
PiperOrigin-RevId: 484561594
1 parent 69d6b8e commit 085f686

File tree

1 file changed

+15
-3
lines changed

1 file changed

+15
-3
lines changed

site/en/install/gpu_plugins.md

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,8 @@ the [Install TensorFlow with pip](./pip.md) guide.
55

66
TensorFlow's
77
[pluggable device](https://github.com/tensorflow/community/blob/master/rfcs/20200624-pluggable-device-for-tensorflow.md){:.external}
8-
architecture adds new device support as separate plug-in packages
9-
that are installed alongside the official TensorFlow package.
8+
architecture adds new device support as separate plug-in packages that are
9+
installed alongside the official TensorFlow package.
1010

1111
The mechanism requires no device-specific changes in the TensorFlow code. It
1212
relies on C APIs to communicate with the TensorFlow binary in a stable manner.
@@ -57,12 +57,24 @@ run() # PluggableDevices also work with tf.function and graph mode.
5757

5858
Metal `PluggableDevice` for macOS GPUs:
5959

60+
* Works with TF 2.5 or later.
6061
* [Getting started guide](https://developer.apple.com/metal/tensorflow-plugin/){:.external}.
6162
* For questions and feedback, please visit the
6263
[Apple Developer Forum](https://developer.apple.com/forums/tags/tensorflow-metal){:.external}.
6364

6465
DirectML `PluggableDevice` for Windows and WSL (preview):
6566

67+
* Works with `tensorflow-cpu` package, version 2.10 or later.
6668
* [PyPI wheel](https://pypi.org/project/tensorflow-directml-plugin/){:.external}.
6769
* [GitHub repo](https://github.com/microsoft/tensorflow-directml-plugin){:.external}.
68-
* For questions, feedback or to raise issues, please visit the [Issues page of `tensorflow-directml-plugin` on GitHub](https://github.com/microsoft/tensorflow-directml-plugin/issues){:.external}.
70+
* For questions, feedback or to raise issues, please visit the
71+
[Issues page of `tensorflow-directml-plugin` on GitHub](https://github.com/microsoft/tensorflow-directml-plugin/issues){:.external}.
72+
73+
Intel® Extension for TensorFlow `PluggableDevice` for Linux and WSL:
74+
75+
* Works with TF 2.10 or later.
76+
* [Getting started guide](https://intel.github.io/intel-extension-for-tensorflow/latest/get_started.html)
77+
* [PyPI wheel](https://pypi.org/project/intel-extension-for-tensorflow/){:.external}.
78+
* [GitHub repo](https://github.com/intel/intel-extension-for-tensorflow){:.external}.
79+
* For questions, feedback, or to raise issues, please visit the
80+
[Issues page of `intel-extension-for-tensorflow` on GitHub](https://github.com/intel/intel-extension-for-tensorflow/issues){:.external}.

0 commit comments

Comments
 (0)