1

要配置 theano,我们在您的主文件夹中创建一个 .theanorc 文件并添加以下内容以设置 theano 以在 GPU 上运行。

[global]
device = gpu
floatx = float32

但有时我会在配置中保存 smbd put device = cuda,有什么区别?据我了解,如果您使用 cuda,它应该工作得更快,因为 cuda 驱动程序将更好地管理 gpu,而不是在计算时为 gpu 获得其他工作

4

1 回答 1

2

device=gpu文件中的设置.theanorc指示 Theano 使用cuda后端。另一方面,设置device=cuda指示 theano 使用libgpuarray后端。两者都导致使用 GPU(与device=cpu.theanorc文件中设置会导致使用 CPU 形成对比),不同之处在于它们用于与之通信的低级 API。后端是两者libgpuarray中较新的一个,它是一个包装器,它允许 Theano 与cuda(对于 NVIDIA GPU)和opencl(对于非 NVIDIA GPU)进行通信。

请注意,cuda后端将在 Theano 的下一个版本中被弃用,建议您libgpuarray以后始终使用后端。

于 2017-02-08T13:57:32.920 回答