не может запустить код горелки на графических процессорах

Question

Я попытался запустить код в факеле учебнике, но когда я хочу запустить его на GPU с помощью :cuda(), я продолжаю получать следующую ошибку

/Opt/zbstudio/bin/Linux/x64/lua: /home/andybug/torch/install/share/lua/5.1/nn/THNN.lua:110: плохой аргумент # 3 to '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: в функции «вперед» ... ug/torch/install/share/lua/5.1/nn/StochasticGradient.lua: 35: в функции 'train' /home/andybug/Documents/НЛП/Lua факел/факел PRAC/main.lua: 74: в основной порции [C]: в 0x00404f08

гугле эту проблему, но никаких действительных ответов, пытался переустановить пп, Cunn, .etc, которое ничем Не работай.

Answer 1

Цель, которую вы передаете ClassNLLCriterion (второй аргумент в прямом направлении), должна быть CudaTensor.

Answer 2

Я столкнулся с той же проблемой при загрузке данных Coco в качестве тендеров Cuda. Поскольку загрузка данных Pascal2012 была прекрасной, я думал, что у меня возникла проблема с загрузчиками данных или итератором (я использую итератор параллельных наборов данных torchnet). При переустановке кусторы я обнаружил, что что-то не так. Затем переустановил Cuda (8 в моем случае), поскольку он сталкивался с другими версиями (7.5) и, наконец, установил факел с нуля. Теперь все работает как шарм.