配置 theano 以使用 libgpuarray

Configuring theano to use libgpuarray

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

device=cuda

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

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

有关更多文档,请参阅 here