Macbook支持GPU深度学习

今天发现了一个可以在苹果本上做深度学习的神器: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

设置环境

1 2	source plaidml-venv/bin/activate plaidml-setup

选择1设置为CPU环境，选择5设置为AMD GPU环境

性能对比

使用plaidbench工具分别测试cpu和gpu下的性能。可以看到GPU比CPU性能提升了10倍左右。