Intel Extension For TensorFlow 发布 – 提供 Intel GPU 加速

Intel 发布了 Intel Extension for TensorFlow,它利用 TF 的 PluggableDevice 机制,现在为 TensorFlow 提供 Intel GPU 后端,可与 Data Center GPU Flex 系列以及 Arc Graphics 离散 GPU 配合使用。

Intel Extension for TensorFlow 现在可以通过 TensorFlow 为 Intel 硬件提供简单的 GPU 加速,补充 NVIDIA CUDA 和 AMD HIP 支持。英特尔支持利用 TensorFlow 的 PluggableDevice 接口来支持这种基于插件的硬件支持,而无需新的 TensorFlow 版本。

image.php_-13

该数据中心 GPU Flex 系列和英特尔 Arc 图形对 TensorFlow 的支持可用于 Linux 以及适用于 Linux 的 Windows 子系统 (WSL2) 并利用英特尔 oneAPI 软件组件。

image.php_-14
适用于 TensorFlow 的英特尔扩展

为 TensorFlow 设置英特尔扩展可以像“ pip install intel-extension-for-tensorflow[gpu] ”一样简单。

原创文章,作者:校长,如若转载,请注明出处:https://www.yundongfang.com/Yun197174.html

(0)
打赏 微信扫一扫不于多少! 微信扫一扫不于多少! 支付宝扫一扫礼轻情意重 支付宝扫一扫礼轻情意重
上一篇 2022年10月29日
下一篇 2022年10月29日

相关推荐