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 在此領域的工作，請訪問

喜歡這篇文章嗎？立刻分享出去讓更多人知道吧！