如何为theano启用MacBook Pro GPU优化?

spe*_*ll4 4 macos gpu theano

  • MacBook Pro(Retina,13英寸,2014年中)
  • OS X Yosemite 10.10.3
  • 英特尔Iris 1536MB

安装Theano显示我需要CUDA,但我没有NVIDIA,这意味着我永远不能启用GPU优化?

Dan*_*haw 6

你是对的:目前需要一个NVIDIA图形处理器来启用Theano的GPU操作.但是,这并不妨碍您运行Theano - 它在CPU上工作得很好.

Theano目前的GPU实现基于CUDA,因此需要NVIDIA GPU.基于OpenCL的新实现正在开发中,应该能够在非NVIDIA GPU上运行,但是这种实现是不完整的,并且还没有实际用处.

CPU实现将正常工作.在许多方面,它比GPU实现更容易使用,如果使用OpenMP,通过使用许多CPU内核仍然可以很好地执行.