python调用gpu amd_如何在AMD的GPU上运行TensorFlow？

工资不涨，英伟达 GPU 的售价年年涨。因此，多一个竞争对手，总是好事。

近日，Google 宣布推出适用于 ROCm GPU 的 TensorFlow v1.8，其中包括 Radeon Instinct MI25。对于 AMD 正在进行的深度学习加速工作而言，这是一座重大的里程碑。

ROCm 即 Radeon 开放生态系统 (Radeon Open Ecosystem)，是我们在 Linux 上进行 GPU 计算的开源软件基础。而 TensorFlow 实现则使用了 MIOpen，这是一个适用于深度学习的高度优化 GPU 例程库。

AMD 提供了一个预构建的 whl 软件包，安装过程很简单，类似于安装 Linux 通用 TensorFlow。目前 Google 已发布安装说明及预构建的 Docker 映像。下面，我们就来手把手地教大家。

▌如何在 AMD 的 GPU 上运行 TensorFlow？

首先，你需要安装开源 ROCm 堆栈，详细的安装说明可以参考：

https：//rocm.github.io/ROCmInstall.html

然后，你需要安装其他相关的 ROCm 软件包：

sudo apt updatesudo apt install rocm-libs miopen-hip cxlactivitylogger

最后，安装 TensorFlow (通过 Google 预先构建的 whl 软件包)：

sudo apt install wget python3-pip wget http://repo.radeon.com/rocm/misc/tensorflow/tensorflow-1.8.0-cp35-cp35m-manylin