【问题标题】:How to speed up GPU mode convolutional neural network with theano?如何用 theano 加速 GPU 模式的卷积神经网络?
【发布时间】:2015-05-16 16:06:48
【问题描述】:

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

程序运行的时候,电脑好像死机了,每次操作真的很慢,甚至没有反应。而且 CPU 模式似乎比 GPU 模式快至少 2 倍。

有没有什么方法可以加快 GPU 卷积神经网络的速度?

【问题讨论】:

    标签: neural-network gpu theano conv-neural-network


    【解决方案1】:

    确保将 Theano 0.7 与 cudnn 一起使用,这会大大加快卷积速度:

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

    【讨论】:

      【解决方案2】:

      要使用 GPU 加速,您首先需要安装 CUDA。 在 Theano 配置级别(Theano flags/TheanoRC)有几种方法可以使用 GPU 加速您的模型:

      1. 指定 GPU “device = gpu”的用法
      2. 启用 Cuda 内存分配 (CnMem) "cnmem = 0.75"
      3. 启用 CUDNN 优化“optimizer = cudnn”

      您可以阅读有关 Theano 配置的更多信息here

      【讨论】:

        猜你喜欢
        • 2016-07-21
        • 2016-03-28
        • 2017-01-01
        • 1970-01-01
        • 1970-01-01
        • 2015-11-15
        • 1970-01-01
        • 1970-01-01
        相关资源
        最近更新 更多