Google 近日宣布：推出适用于 AMD ROCm GPU 的 TensorFlow v1.8，其中包括 Radeon Instinct MI25。对于 Google 正在进行的深度学习加速工作而言，这是一座重大的里程碑。ROCm 即 Radeon 开放生态系统 (Radeon Open Ecosystem)，是在 Linux 上进行 GPU 计算的开源软件基础。AMD 的 TensorFlow 实现使用了 MIOpen，这是一个适用于深度学习的高度优化 GPU 例程库。AMD 提供了一个预构建的 whl 软件包，安装过程很简单，类似于安装 Linux 通用 TensorFlow。同时，已发布安装说明及预构建的 Docker 映像。
项目 GitHub 地址