如何在Mac OS X EI Capitan下安装带有OpenCL支持的Theano

我有两台Mac,一台Mac pro和一台Macbook pro。他们都是用AMD,所以我无法使用CUDA加速机器学习,所以,我想使用OpenCL。

首先安装一些必备的库和工具:

然后建立一个安装环境 (使用virtualenv可以避免很多麻烦的问题):

安装Theano:

然后,你可以用这个测试代码 ( test.py ) 测试Theano的安装:

这测试会显示如下的信息:

其实就是做一些数学运算做1000次,花了1.49225秒,它在使用CPU。

然后我们必须安装libgpuarray,这样Theano才能支持OpenCL:

然后,我们可以用THEANO_FLAGS设置我们需要用的OpenCL设备。

在我的Mac pro使用OpenCL和CPU:

注意: 代码显示“Used the gnu”,但是实际上你能看出这是CPU。

在我的Mac pro使用OpenCL和GPU:

终于,我们可以使用OpenCL和GPU了。

性能:

我的Mac pro

屏幕截图 2016-04-01 15.09.25

我的Macbook pro

屏幕截图 2016-04-01 15.09.32

注意:现在我们可以用Theano和OpenCL了。但是,当我发现Theano使用CUNN的时候,我很悲哀,因为当你用Theano做神经网络训练的时候,它仍旧需要CUDA的支持。暂时,我还没办法解决这个问题。

英文版本:How to install Theano on Mac OS X EI Caption with OpenCL support