torch / rnn不会使用CUDA

时间:2017-10-21 12:08:54

标签: torch rnn

我正在尝试使用torch/rnn工具包在我的nVidia显卡上运行RNN。我有一个带有nVidia驱动程序,CUDA工具包,Torch和cuDNN的Ubuntu 16.04虚拟机。我可以运行mnistCUDNN示例,nvidia-smi使用图形卡显示它。在火炬,我可以要求('cunn');并且它快乐地装载。

但是当我dofile('。/ rnn / examples / recurrent-visual-attention.lua');在火炬里面,我得到了

{
   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运行。

任何想法如何解决我错过的东西?感谢。

1 个答案:

答案 0 :(得分:0)

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

在我的辩护中,这些例子没有记载......