在AMD GPU中使用Python + Theano和OpenCL

Question

我正在尝试使用Python与Theano来加速OpenCL的一些代码.我安装libgpuarray并pygpu按照指示(我认为),并没有错误.安装检测到安装的OpenCL运行时.

我只是不能为OpenCL运行Theano示例,主要是因为我不知道如何指定我的GPU.据我所知Radeon HD 5340/5450/5470,我的GPU是一款inxi.Theano文档中的所有代码都是使用的device=cuda0,也是唯一提到OpenCL的地方,它说的是数字device=openclN在哪里N.

我尝试device=opencl0了一个pygpu错误,说正确的格式是opencl<int>:<int>.我已经尝试过任意数量的数字组合(opencl0:0等等),而且总是一个GpuArrayException: Unknown error.

我的系统是Ubuntu 14.04 x64,我的硬件是东芝Satellite,15".我安装Theano了pip,后来libgpuarray按照他们网站上的说明安装.

我究竟做错了什么？

Answer 1

opencl0:0是正确的.你能证实pyopencl有效吗？您的opencl(或驱动程序/ cl编译器)可能有问题.

但是,我认为Theano目前与OpenCL并不完全一致.目前的状态是有部分支持,足够"hello world",但不足以运行任何重要的代码.

看到:

https://github.com/Theano/Theano/issues/2189

https://github.com/Theano/Theano/issues/1471

https://github.com/Theano/Theano/issues/2190

https://github.com/Theano/Theano/pull/1732

总而言之,不,大多数东西都没有被移植(例如,包括Elemwise,这是一个常见的操作).我真的很想在OpenCL上看到Theano.对于AMD而言,这将是一件好事.很快:)