1. 程式人生 > >AMD ROCm GPU,支援 TensorFlow!

AMD ROCm GPU,支援 TensorFlow!

文 / 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/deeplearning