https://blogger.googleusercontent.com/img/b/R29vZ2xl/AVvXsEjXl8SHNDNVT2N8jx9RlJcZDFnSgK5sgA17Zsf_xmeX6a5HdYk5NgWu5_2TWah8PvIKX65yvJFG1apPSCjX3J4cvGErdfrPj-JtNyWzRzQwNj9bVnt3jd2TDzwcUfF1efkosuhXy3M9SHg/s1600/amd.png
AMD 深度学习软件总监 Mayank Daga 的客座文章
我们很高兴地宣布发布支持 ROCm 启用的 GPU(包括 Radeon Instinct MI25)的 TensorFlow v1.8。这是 AMD 在加速深度学习方面持续努力取得的重大里程碑。
ROCm(Radeon 开放生态系统)是我们在 Linux 上用于 GPU 计算的开源软件基础。我们的 TensorFlow 实现利用了
MIOpen,这是一个针对深度学习的高度优化的 GPU 例程库。
AMD 提供了预构建的
whl 包,允许进行简单的安装,类似于安装用于 Linux 的通用 TensorFlow。我们已经发布了
安装说明,以及
预构建的 Docker 镜像。
除了支持 TensorFlow v1.8 之外,我们还致力于将所有 ROCm 特定的增强功能上游到 TensorFlow 主存储库。其中一些补丁已经合并到上游,而更多补丁正在积极审查中。在我们努力完全将增强功能上游的同时,我们将发布和维护未来的 ROCm 启用的 TensorFlow 版本,例如 v1.10。
我们相信深度学习优化、可移植性和可扩展性的未来根植于特定领域的编译器。我们受到 XLA 早期成果的激励,并且也致力于为 AMD GPU 启用和优化 XLA。
有关 AMD 在该领域工作的更多信息,请参阅
www.amd.com/deeplearning