在 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.


支付宝打赏 微信打赏

您的打赏是对我最大的鼓励!

linzhanyu


Linux, VIM, OpenGL, Unity3D, AI.

相看莫相笑,同是竹林人。