Using Python+Theano with OpenCL in an AMD GPU

后端 未结 1 1595
天命终不由人
天命终不由人 2021-02-02 14:59

I\'m trying to use Python with Theano to accelerate some code with OpenCL. I installed libgpuarray and pygpu as instructed (I think), and got no errors

1条回答
  •  自闭症患者
    2021-02-02 15:36

    opencl0:0 is correct. Could you confirm that pyopencl works? You may have a problem with your opencl (or drivers/cl compiler).

    However, I think Theano does not quite work with OpenCL at the moment. The current state is there is partial support, enough for "hello world", but not enough to run any significant code.

    See:

    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

    To summarize, no, most stuff is not ported (including Elemwise, for example, which is a common op). I would really like to see Theano on OpenCL though. That would be a great thing for AMD to pitch in on. Soon :)

    0 讨论(0)
提交回复
热议问题