torch/rnn不会使用CUDA

问题描述：

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

但是当我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：假的，它运行只使用CPU。

任何想法如何解决我错过了什么？谢谢。

答

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

在我的辩护中，这些例子是无证的...

torch/rnn不会使用CUDA

相关推荐