AMD ROCm GPU,支持 TensorFlow!

Posted TensorFlow

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了AMD ROCm GPU,支持 TensorFlow!相关的知识,希望对你有一定的参考价值。

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


我们很高兴地宣布推出适用于 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/deeplearning


更多 AI 相关阅读:

 


点击 “阅读原文”,进一步了解 AMD!

以上是关于AMD ROCm GPU,支持 TensorFlow!的主要内容,如果未能解决你的问题,请参考以下文章

AMD GPU 上的 CuPy 导致 ImportError

即使我们有单节点多 GPU 设置,也限制 MPI 在单 GPU 上运行

Running Tensorflow on AMD GPU

将 Keras 和 Tensorflow 与 AMD GPU 结合使用

将 OSX 上的 OpenCL 设备设置为专用的 AMD GPU

cuda11.7显卡要求