tensorflow在AMD上用ROCm框架实现GPU训练（亲测有效）

众所周知，a卡对于tensorflow gpu的支持很小，要想使用的话一般都得配n卡的电脑。而我只是感兴趣一个项目，想要跑出来一个模型，如果不用gpu加速，可能得跑一个月，所以不得不寻找在a卡上的解决方案。好在，终于找到了一个解决方案，虽然只是取巧，速度也不如正常n卡跑的快，但已经很不错了。

我的配置：
显卡：RX 550
所用系统：虚拟机ubuntu 18
注：如果是虚拟机的话，最好分配3GB内存或以上，否则可能无法import tensorflow

a卡上支持的ROCm框架，暂时只支持ubuntu系统。如果你是windows系统，也可以安一个虚拟机，道理是类似的，我也是在虚拟机上跑的。用虚拟机进行下面教程的时候：会遇到无法导入显卡信息的bug，但是不影响训练情况。经过测试，我在虚拟机上训练的速度是之前在a卡上训练速度的四五倍，不知道安linux子系统是不是会再快一点，但我没做尝试。要注意，你的AMD显卡是否支持ROCm框架，请参考以下链接确定你的显卡是否支持。https://rocm.github.io/hardware.html 。还有你的工程如果不是python3.5/3.6,你也用不了此框架。

如果条件都符合，就可以开始愉快的安装了。（变通的法子就是费劲儿。。。）

更新系统

建议：把源换为国内的，比如阿里的，会快很多的。

sudo apt update
sudo apt upgrade

安装ROCm

添加ROCm库（没有