python调用gpu amd_TensorFlow通过AMD GPU加速（ROCm/elementary OS 5.0）

elementary OS 5.0是基于Ubuntu 18.04制作，所以本流程也适用于Ubuntu 18.04。

ROCm升级1.91之后不需要安装AMD的GPU驱动程序，如果安装可能会出现问题。

安装elementary OS 5 之后的一些配置

安装软件更新管理器

sudo apt update

sudo apt install software-properties-common software-properties-gtk

运行软件更新管理器然后选择合适的镜像，我使用的是阿里云。

sudo software-properties-gtk

更新系统

sudo apt update

sudo apt upgrade

安装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 -y \

build-essential \

clang \

clang-format \

clang-tidy \

cmake \

cmake-qt-gui \

ssh \

curl \

apt-utils \