10 月 31 日消息,TensorFlow 是一个端到端开源机器学习平台。它拥有一个全面而灵活的生态系统,其中包含各种工具、库和社区资源,借助 TensorFlow,初学者和专家可以轻松创建适用于桌面、移动、Web 和云环境的机器学习模型。
随着英特尔自家锐炫独显的发布,英特尔官方宣布推出 Intel Extension for TensorFlow,这是一种在英特尔 AI 硬件上运行 TensorFlow 应用的开源解决方案。
据介绍,Intel Extension for TensorFlow 是一个高性能深度学习扩展,实现了 TensorFlow PluggableDevice 接口。通过与 TensorFlow 框架的无缝集成,它允许 TensorFlow 开发人员轻松访问英特尔 XPU(GPU、CPU 等)设备。借助英特尔扩展,开发人员可以在零代码更改的情况下在英特尔 AI 硬件上训练和推断 TensorFlow 模型。
Intel Extension for TensorFlow 构建在 oneAPI 软件组件之上。大多数性能关键图和运算符都通过使用英特尔 oneAPI 深度神经网络 (oneDNN) 进行了高度优化,这是一个用于深度学习应用程序的开源跨平台性能库。其他运算符使用 SYCL 实现,这是一种用于编程加速器和多处理器的 API 核心语言。
▲ 外媒 phoronix 进行了测试
该插件还维护一个本地 Eigen 数学库,该库将 Eigen 移植到 SYCL 语言,以便它可以生成 SYCL 内核程序来实现设备运算符。Intel Extension for TensorFlow 还实现了所有 TensorFlow GPU 运算符,从而确保英特尔 GPU 具有良好的性能和足够的模型覆盖率。
IT之家了解到,Intel Extension for TensorFlow 可搭配 Flex 系列数据中心 GPU 和英特尔锐炫系列用于 Linux 以及 Windows 11 的 Linux 子系统 (WSL2) 。