AMD ROCm GPU,支持 TensorFlow!

1,083 阅读1分钟

文 / AMD 深度学习软件总监 Mayank Daga

来源 | TensorFlow 公众号

我们很高兴地宣布推出适用于 ROCm GPU 的 TensorFlow v1.8,其中包括 Radeon Instinct MI25。对于 AMD 正在进行的深度学习加速工作而言,这是一座重大的里程碑。ROCm 即 Radeon 开放生态系统 (Radeon Open Ecosystem),是我们在 Linux 上进行 GPU 计算的开源软件基础。我们的 TensorFlow 实现使用了 MIOpen,这是一个适用于深度学习的高度优化 GPU 例程库。

AMD 提供了一个预构建的 whl 软件包,安装过程很简单,类似于安装 Linux 通用 TensorFlow。我们已发布安装说明及预构建的 Docker 映像。

除了支持 TensorFlow v1.8 以外,我们还在努力向 TensorFlow 主存储区加入所有针对 ROCm 的增强功能。其中一些补丁程序已经在上游合并,另外几个正在积极审查中。在努力全面加入增强功能的同时,我们还将发布和维护日后支持 ROCm 的 TensorFlow 版本,例如 v1.10。

我们相信深度学习优化、可移植性和可伸缩性的未来根植于特定领域的编译器。我们受到 XLA 早期结果的启发,也在致力于为 AMD GPU 启用和优化 XLA。

如需进一步了解 AMD 在此领域的工作,请访问 www.amd.com/deeplearnin…