2016-09-30 92 views
1

火炬代碼我試圖運行在火炬教程中的代碼,但是當我想使用:cuda() GPU運行它,我不斷收到以下錯誤不能運行在GPU

的/ opt/zbstudio /斌/ Linux的/ x64/lua:/home/andybug/torch/install/share/lua/5.1/nn/THNN.lua:110:錯誤的參數#3到'v'(不能將'struct THCudaTensor *'轉換爲'struct THCudaLongTensor * 「) 棧回溯: [C]:在函數 'v' /home/andybug/torch/install/share/lua/5.1/nn/THNN.lua:110:在函數 'ClassNLLCriterion_updateOutput' ...錯誤/手電筒/安裝/股/ LUA/5.1/nn/ClassNLLCriterion.lua:41:在函數'forward'中 ... ug/torch/install/share/lua/5.1/nn/StochasticGradient.lua:35:函數'train' /home/andybug/Documents/NLP/LUA手電筒/火炬PRAC/main.lua:74:主塊 [C]:在0x00404f08

Google搜索這個問題,但沒有有效的答案,嘗試重新安裝NN,cunn,.etc,這沒沒有工作。

回答

-1

您傳遞給ClassNLLCriterion(轉發中的第二個參數)的目標需要是CudaTensor。

0

加載Coco數據爲Cuda張量時,我遇到過同樣的問題。由於加載Pascal2012數據沒有問題,我以爲我遇到了數據加載器或迭代器的問題(我正在使用torchnet的並行數據集迭代器)。在重新安裝cutorch我發現有什麼不對。然後重新安裝Cuda(我的情況是8),因爲它與其他版本(7.5)相撞,最後從頭開始安裝火炬。現在一切都像一個魅力。