Theano设置.theanorc文件不同于gpu和cuda之间

时间:2017-02-04 22:16:34

标签: gpu theano theano-cuda

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

[global]
device = gpu
floatx = float32

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

1 个答案:

答案 0 :(得分:2)

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

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