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。

任何想法如何解決我錯過了什麼？謝謝。

來源

2017-10-21 TomG

我是個白癡。當我終於鼓起勇氣閱讀源代碼時，我發現它不會自動嘗試使用CUDA。有一個-cuda標誌要求它。

在我的辯護中，這些例子是無證的...

來源

2017-10-22 11:12:27 TomG

torch/rnn不會使用CUDA

回答

相關問題