英特尔® TensorFlow 扩展

英特尔® Extension for TensorFlow* 是一个基于 TensorFlow PluggableDevice接口的异构、高性能深度学习扩展插件,可将英特尔 XPU(GPU、CPU 等)设备引入TensorFlow开源社区,以实现 AI 工作负载加速。它允许灵活地按需将 XPU 插入 TensorFlow,并在英特尔硬件中公开计算能力。

项目地址:https://github.com/intel/intel-extension-for-tensorflow


此图提供了 TensorFlow* PyPI 包生态系统的摘要。

  • TensorFlow PyPI 包:estimatorkerastensorboardtensorflow-base
  • 英特尔® TensorFlow* 扩展包:
  • intel_extension_for_tensorflow包含:
  • XPU 具体实现
  • 内核和运算符
  • 图优化器
  • 设备运行时间
  • XPU 配置管理
  • XPU 后端选择
  • 打开/关闭高级功能的选项

安装

硬件要求

英特尔® Extension for TensorFlow* 提供英特尔 GPU支持和实验性英特尔 CPU支持。

软件要求

包裹 中央处理器 图形处理器 安装 英特尔 GPU 驱动程序 是 安装英特尔 GPU 驱动程序 英特尔® oneAPI 基础工具包 是 安装英特尔® oneAPI 基础工具包 TensorFlow 是 是 安装 TensorFlow 2.10.0

安装渠道:

可从以下渠道安装英特尔® Extension for TensorFlow*:

派皮 码头工人中心 资源 显卡\中央处理器 GPU容器\ CPU容器 从源代码构建

为 GPU 安装

pip install tensorflow==2.10.0
pip install --upgrade intel-extension-for-tensorflow[gpu]

有关详细信息,请参阅 GPU 安装

安装 CPU [实验]

pip install tensorflow==2.10.0
pip install --upgrade intel-extension-for-tensorflow[cpu]

健全性检查:

python -c "import intel_extension_for_tensorflow as itex; print(itex.__version__)"

文档

访问在线文档网站,然后开始浏览英特尔® Extension for TensorFlow*示例

贡献

我们欢迎社区为英特尔® Extension for TensorFlow* 做出贡献。

该项目旨在成为一个安全、欢迎协作的空间,贡献者应遵守贡献者公约。请参阅贡献指南了解更多详情。

资源

支持

请在GitHub 问题页面上提交您的问题、功能请求和错误报告。

安全

有关如何报告潜在安全问题或漏洞的信息,请参阅英特尔安全中心。

另请参阅:安全策略

执照

阿帕奇许可证 2.0

此分发包括受单独许可条款约束的第三方软件。此第三方软件,即使包含在英特尔软件的分发中,也可能受单独的许可条款的约束,包括但不限于第三方许可条款、其他英特尔软件许可条款和开源软件许可条款。这些单独的许可条款约束您对“第三方程序”文件中规定的第三方程序的使用。

© GVGNN 2013-2026