Plaidml--MAC+A卡的深度学习方案

Plaidmi 适合所有显卡的深度学习加速平台

特性

• 支持Kears（定制版）

• 支持A卡

• 基于OpenCL加速

安装

安装环境：macOS 10.15.3(OpenCL1.2)/Anconda3/Python3.7

1. 安装

1.1 安装plaidml-keras

安装官网使用pip安装，但最好指定较低版本，高版本亲测会有莫名的bug。

pip install plaidml-keras==0.6.4

1.2 设置默认配置（选择加速优先使用的硬件）

2 demo

只比正常的kears代码多行代码

os.environ["KERAS_BACKEND"] = "plaidml.keras.backend"

3 和pytorch对比

显卡不同，不好比较，实验室的N卡是GTX 970 显存4G， 笔记本显卡A卡为Radeon Pro 555X显存4G

DnCNN（patchsize=40x40, batch_size=128, iters=2000）训练1个epoch所需时间分别为1个小时和1.6个小时左右。

4 总结

由于此次疫情，返校时间一推再推，然而大论文提交时间并不会随之延迟。原本计划在家写好大论文的文字部分，等3月开学再补实验。可现实是：并没有如期写完文字部分和返校时间再次延迟。

今天早上起来我一度想买张显卡给家里的台式装上，还好我的口袋并不充裕，暂时打消了这个念头。最后无意中在b站看到标题为”如何在MAC下使用AMDGPU跑神经网络“的科普视频（还以为是标题党）。。。

随后按照plaidml官网安装，跑示例demo没问题，运行自己搭建的网络确发现loss一直为NAN异常值，而用cpu确正常。本来遇到bug再正常不过了，可难顶的是这玩意用的人特别少，相关博客都搜不到几篇。。。

无计可施想问问大佬，可这么冷门大佬估计也没用过，也不好麻烦人家。

几经折腾无果，晚饭后用手机翻到一篇17分钟前发的热乎的博客，正好是讲如何安装plaidml，并重点强调了要指定低版本安装。。。

重装pliaidmi-kears，运行，loss终于正常。

太折腾了，记录下，给需要的人也给自己有个参考。

完