torch/rnn 不会使用 CUDA答案

【问题标题】：torch/rnn won't use CUDAtorch/rnn 不会使用 CUDA
【发布时间】：2018-04-02 10:47:50
【问题描述】：

我正在尝试使用 torch/rnn 工具包在我的 nVidia 显卡上运行 RNN。我有一个运行 nVidia 驱动程序、CUDA 工具包、Torch 和 cuDNN 的 Ubuntu 16.04 VM。我可以运行 mnistCUDNN 示例，然后 nvidia-smi 使用显卡显示它。在 Torch 中，我可以 require('cunn');并且加载愉快。

但是当我 dofile('./rnn/examples/recurrent-visual-attention.lua' );在 Torch 中，我得到了

{
   batchsize : 20
   cuda : false
   cutoff : -1
   dataset : "Mnist"
   device : 1
   earlystop : 200
   glimpseDepth : 1
   glimpseHiddenSize : 128
   glimpsePatchSize : 8
   glimpseScale : 2
   hiddenSize : 256
   id : "ptb:brain:1508585440:1"
   imageHiddenSize : 256
   locatorHiddenSize : 128
   locatorStd : 0.11
   lstm : false
   maxepoch : 2000
   maxnormout : -1
   minlr : 1e-05
   momentum : 0.9
   noTest : false
   overwrite : false
   progress : false
   rewardScale : 1
   saturate : 800
   savepath : "/home/tom/save/rmva"
   seqlen : 7
   silent : false
   startlr : 0.01
   stochastic : false
   trainsize : -1
   transfer : "ReLU"
   uniform : 0.1
   unitPixels : 13
   validsize : -1
   version : 13
}

由于 cuda:false，它只使用 CPU 运行。

有什么想法可以解决我错过的问题吗？谢谢。

【问题讨论】：

标签： torch rnn

【解决方案1】：

我是个白痴。当我终于鼓起勇气阅读源代码时，我发现它不会自动尝试使用 CUDA。有一个 -cuda 标志可以要求它。

在我的辩护中，这些例子没有记录......

【讨论】：