如何加速与theano的GPU模式卷积神经网络？

Question

我正在使用theano来实现卷积神经网络。我的CPU RAM为32G，GPU RAM为2G，但数据也非常大 - 几乎是5G的训练数据。

当程序运行时，计算机似乎被冻结，每个操作都很慢，甚至没有响应。并且CPU模式似乎比GPU模式快至少2倍。

有没有办法加速GPU卷积神经网络？

Answer 1

确保使用Theano 0.7 with cudnn，这会加速卷积：

http://deeplearning.net/software/theano/library/sandbox/cuda/dnn.html

Answer 2

为了首先使用GPU加速，你需要安装CUDA。 在Theano配置（Theano flags / TheanoRC）的层面上，使用GPU加速模型的方法很少：

1. 指定GPU＆＃34; device = gpu＆＃34;
的用法
2. 启用Cuda内存分配（CnMem）＆＃34; cnmem = 0.75＆＃34;
3. 启用CUDNN优化＆＃34; optimizer = cudnn＆＃34;

您可以阅读有关Theano config here

的更多信息