配置 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。
我正在尝试将 theano 切换到 gpuarray
后端。
这个 tutorial 说要使用 cuda
后端我写
device=cuda
在.theanorc
但它没有指定要写什么才能使用gpuarray
,这是我的问题。
使用 gpuarray
后端只允许您同时针对 OpenCL 和 CUDA 设备,因此您仍然需要在 .theanorc
文件中明确针对 CUDA 或 OpenCL 设备。在引擎盖下,它仍在使用 OpenCL 或 CUDA,因此设备目标保持不变,gpuarray
只是提供了一个额外的抽象层,允许使用其中任何一个。
有关更多文档,请参阅 here。