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 版本。
该数据中心 GPU Flex 系列和英特尔 Arc 图形对 TensorFlow 的支持可用于 Linux 以及适用于 Linux 的 Windows 子系统 (WSL2) 并利用英特尔 oneAPI 软件组件。
适用于 TensorFlow 的英特尔扩展
为 TensorFlow 设置英特尔扩展可以像“ pip install intel-extension-for-tensorflow[gpu] ”一样简单。
原创文章,作者:校长,如若转载,请注明出处:https://www.yundongfang.com/Yun197174.html