在AMD GPU中使用Python+Theano和OpenCL

2024-06-28 20:15:01 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在尝试使用Python和Theano来使用OpenCL加速一些代码。我按照指示(我想)安装了libgpuarraypygpu,没有错误。安装程序检测到已安装OpenCL运行时。

我不能运行OpenCL的Theano示例,主要是因为我不知道如何指定我的GPU。根据inxi,我的GPU是Radeon HD 5340/5450/5470。Theano文档中的所有代码都使用device=cuda0,而提到OpenCL的唯一地方是device=openclN,其中N是一个数字。

我尝试了device=opencl0,得到一个pygpu错误,说正确的格式是opencl<int>:<int>。从那以后,我尝试过任意数量的数字组合(opencl0:0等等),并且总是一个GpuArrayException: Unknown error

我的系统是Ubuntu14.04x64,我的硬件是东芝卫星,15“。我用pip安装了Theano,然后按照他们站点上的说明安装了libgpuarray

我做错什么了?


Tags: 代码示例gpudevice错误数字theanoint
1条回答
网友
1楼 · 发布于 2024-06-28 20:15:01

opencl0:0是正确的。你能确认一下pyopencl的工作原理吗?您的opencl(或drivers/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投入。很快:)

相关问题 更多 >