rocm平台_TensorFlow通过AMD GPU加速（ROCm/Ubuntu 18.04）

ROCm 1.91之后不需要安装AMD GPU驱动程序。请参考新的安装流程：

通过AMD开发ROCm平台，TensorFlow可以使用AMD GPU实现GPU加速。现将搭建流程呈上。

硬件：

CPU：AMD Ryzen 1700x

GPU：AMD Radeon RX580

内存：32G

硬盘：SSD 256GB + HDD 2TB

安装Ubuntu 18.04

网上很多Ubuntu安装教程，这里不在赘述。我选的是最小化安装。

安装AMD GPU驱动程序

下载最新的驱动程序，我使用的是18.20版本。

以下载到Downloads目录为例

cd ~/Downloads

tar -Jxvf amdgpu-pro-18.20-606296.tar.xz

cd ~/Downloads/amdgpu-pro-18.20-606296

./amdgpu-pro-install –opencl=legacy

安装ROCm

增加ROCm的仓库

wget -qO - http://repo.radeon.com/rocm/apt/debian/rocm.gpg.key | sudo apt-key add -

sudo sh -c 'echo deb [arch=amd64] http://repo.radeon.com/rocm/apt/debian/ xenial main > /etc/apt/sources.list.d/rocm.list'

然后运行

sudo apt update

sudo apt install rocm-dkms

<