2016-09-30 87 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)相撞,最后从头开始安装火炬。现在一切都像一个魅力。