我正在尝试使用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运行。
任何想法如何解决我错过的东西?感谢。
答案 0 :(得分:0)
在我的辩护中,这些例子没有记载......