Theano + OpenCL

在 Linux 下要尝试搞一搞机器学习，可惜只有一张A卡，就让我用 OpenCL 来加速吧。

找了找当下的机器学习框架 介绍几个支持 OpenCL 为后端加速的安装方法吧

Theano

这里需要注意的是

• libgpuarray 用源码编译时要取 >= 0.7.0 的 TAG，编译时要加相应的 Lib 目录，要把 python 指向 python3 以方便内部的命令调用
• Theano 用源码安装

测试

1. 测试 libgpuarray

DEVICE=opencl0:0 python3 -c "import pygpu; pygpu.test()"

1. 测试 Theano

THEANO_FLAGS=mode=FAST_RUN,DEVICE=opencl0:0,floatX=float32 python3 testTheanoDevice.py

一些自定义配置

Theano 配置超方便 只需要编辑 ~/.theanorc 即可实现

使用 GPU

使用 clblast 做为调用的 blas

[blas]
ldflags = -lclblast

Keras

Keras -> PlaidML -> OpenCL

其它计算

其它的矩阵计算准备使用 clBlast 来搞，对应的 Python 调用 Bind 是 pyclblast。编译安装 SO EASY.