【问题标题】:cannot run torch code on gpu无法在 GPU 上运行 Torch 代码
【发布时间】:2017-02-08 20:47:14
【问题描述】:

我尝试在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]:在函数“v”中 /home/andybug/torch/install/share/lua/5.1/nn/THNN.lua:110:在函数“ClassNLLCriterion_updateOutput”中 ...bug/torch/install/share/lua/5.1/nn/ClassNLLCriterion.lua:41:在函数“转发”中 ...ug/torch/install/share/lua/5.1/nn/StochasticGradient.lua:35:在函数'train'中 /home/andybug/Documents/NLP/lua torch/torch-prac/main.lua:74:在主块中 [C]: 在 0x00404f08

搜索了这个问题,但没有有效的答案,尝试重新安装 nn、cunn、.etc,但没有成功。

【问题讨论】:

    标签: lua torch


    【解决方案1】:

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

    【讨论】:

      【解决方案2】:

      我在将 Coco 数据加载为 Cuda 张量时遇到了同样的问题。由于加载 Pascal2012 数据很好,我认为我遇到了数据加载器或迭代器的问题(我正在使用 torchnet 的并行数据集迭代器)。在重新安装 cutorch 时,我发现有问题。然后重新安装 Cuda(在我的情况下为 8),因为它与其他版本(7.5)发生冲突,最后从头开始安装 Torch。现在一切都像魅力一样运作。

      【讨论】:

        猜你喜欢
        • 1970-01-01
        • 2011-05-06
        • 2017-07-27
        • 2018-04-12
        • 2018-09-11
        • 1970-01-01
        • 1970-01-01
        • 1970-01-01
        相关资源
        最近更新 更多