配置theano以使用libgpuarray

时间:2017-07-12 13:48:31

标签: python config theano

我正在尝试将theano切换到gpuarray后端。 这个tutorial表示要使用cuda后端我写

device=cuda

.theanorc 但它没有指定要使用gpuarray写什么,这是我的问题。

1 个答案:

答案 0 :(得分:1)

使用gpuarray后端只允许您定位OpenCL和CUDA设备,因此您仍需要在.theanorc文件中明确定位CUDA或OpenCL设备。在引擎盖下,它仍然使用OpenCL或CUDA,因此设备目标保持不变,gpuarray只是提供了一个额外的抽象层,允许使用它们。

有关详细文档,请参阅here