要配置 theano,我们在您的主文件夹中创建一个 .theanorc 文件并添加以下内容以设置 theano 以在 GPU 上运行。
[global]
device = gpu
floatx = float32
但有时我会在配置中保存 smbd put device = cuda,有什么区别?据我了解,如果您使用 cuda,它应该工作得更快,因为 cuda 驱动程序将更好地管理 gpu,而不是在计算时为 gpu 获得其他工作
要配置 theano,我们在您的主文件夹中创建一个 .theanorc 文件并添加以下内容以设置 theano 以在 GPU 上运行。
[global]
device = gpu
floatx = float32
但有时我会在配置中保存 smbd put device = cuda,有什么区别?据我了解,如果您使用 cuda,它应该工作得更快,因为 cuda 驱动程序将更好地管理 gpu,而不是在计算时为 gpu 获得其他工作
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
以后始终使用后端。