今天发现了一个可以在苹果本上做深度学习的神器:plaidml。基本上所有的macbook苹果本都是使用的AMD显卡,不支持Navid显卡,所以苹果笔记本的Tensorflow等机器学习框架不能在GPU上执行。plaidml支持不同类型的GPU,并提供了plaidml-keras使得用户可以直接用keras编程不用关心后面的GPU是什么类型。
安装过程
首先保证已经安装了keras和tensorflow。
python3 -m venv plaidml-venv
source plaidml-venv/bin/activate
pip install -U plaidml-keras
pip install plaidml-keras plaidbench
设置环境
source plaidml-venv/bin/activate
plaidml-setup
![6b3ade436651ef13b55e3b9f647528e0.png](https://i-blog.csdnimg.cn/blog_migrate/b64c09d56bbd04ea749d13afae9863e2.jpeg)
选择1设置为CPU环境,选择5设置为AMD GPU环境
性能对比
使用plaidbench工具分别测试cpu和gpu下的性能。
![c94fbe4636a8a0eea721096f38f9abe3.png](https://i-blog.csdnimg.cn/blog_migrate/d4adc5f4c36e98575f5aae069d15ddec.jpeg)
![28e4bac4764977b187aa97c2372c5362.png](https://i-blog.csdnimg.cn/blog_migrate/d76001a482c42629646a39c8aa37cab2.jpeg)
可以看到GPU比CPU性能提升了10倍左右。