无法在gpu上运行火炬代码

Question

我试图在torch教程中运行代码，但是当我想使用:cuda()在gpu上运行它时，我一直收到以下错误

/opt/zbstudio/bin/linux/x64/lua: /home/andybug/torch/install/share/lua/5.1/nn/THNN.lua:110：坏参数#3到'v‘(不能将'struct THCudaTensor *’转换为'struct THCudaLongTensor *')堆栈回溯: C: in函数'v‘/home/andybug/torch/install/share/lua/5.1/nn/THNN.lua:110: in function 'ClassNLLCriterion_updateOutput’...bug/torch/install/share/lua/5.1/nn/ClassNLLCriterion.lua:41: in函数‘前进’...ug/torch/install/share/lua/5.1/nn/StochasticGradient.lua:35: in function 'train‘/home/andybug/Documents/NLP/lua手电筒/手电筒-prac/main.lua:74:主块C: 0x00404f08

谷歌搜索了这个问题，但没有得到有效的答案，于是尝试重新安装nn，cunn，.etc，这是无效的。

Answer 1

在加载Coco数据时，我遇到了与Cuda张量相同的问题。由于加载Pascal2012数据很好，我认为数据加载器或迭代器存在问题(我使用的是torchnet的并行数据集迭代器)。在重新安装割炬时，我发现有什么地方不对劲。然后重新安装Cuda (在我的例子中是8)，因为它与其他版本(7.5)发生冲突，最后从零开始安装火炬。现在一切都像魅力一样运作。

Answer 2

传递给ClassNLLCriterion (前向的第二个参数)的目标需要是一个CudaTensor。