run_model.lua fails for me when I run in GPU mode: <div class="highlight highlight

Try upgrading cudnn and nngraph: <div class="highlight highlight-source-shell notr

<a class="user-mention notranslate" data-hovercard-type="user" data-hovercard-url="/us

Ah found <a class="issue-link js-issue-link" data-error-text="Failed to load title" da

run_model.lua fails with "cudnn.convert does not work with nngraph yet" about densecap HOT 4 CLOSED

jcjohnson commented on June 3, 2024

run_model.lua fails with "cudnn.convert does not work with nngraph yet"

from densecap.

Comments (4)

jcjohnson commented on June 3, 2024

Try upgrading cudnn and nngraph:

luarocks install cudnn
luarocks install nngraph

from densecap.

soumith commented on June 3, 2024

@snowmonkeypc updating cudnn will fix it via:

luarocks install cudnn

from densecap.

snowmonkeypc commented on June 3, 2024

Thanks! "luarocks install cudnn" turned the error into a warning. Is that expected?

I'm now stuck with an "out of memory" error. Only 2 GB VRAM in this machine (and probably much less available even directly after reboot), guess it's simply not enough.

$ th run_model.lua -input_image imgs/elephant.jpg
Warning: cudnn.convert does not work with nngraph yet. Ignoring nn.gModule1/1 processing image imgs/elephant.jpg
THCudaCheck FAIL file=/tmp/luarocks_cutorch-scm-1-4722/cutorch/lib/THC/generic/THCStorage.cu line=40 error=2 : out of memory
/Users/snowm/torch/install/bin/luajit: /Users/snowm/torch/install/share/lua/5.1/nn/Container.lua:67:
In 4 module of nn.Sequential:
In 5 module of nn.Sequential:
/Users/snowm/torch/install/share/lua/5.1/nn/Linear.lua:61: cuda runtime error (2) : out of memory at /tmp/luarocks_cutorch-scm-1-4722/cutorch/lib/THC/generic/THCStorage.cu:40
stack traceback:
        [C]: in function 'resize'
        /Users/snowm/torch/install/share/lua/5.1/nn/Linear.lua:61: in function </Users/snowm/torch/install/share/lua/5.1/nn/Linear.lua:53>
        [C]: in function 'xpcall'
        /Users/snowm/torch/install/share/lua/5.1/nn/Container.lua:63: in function 'rethrowErrors'
        /Users/snowm/torch/install/share/lua/5.1/nn/Sequential.lua:44: in function 'func'
        ...rs/snowm/torch/install/share/lua/5.1/nngraph/gmodule.lua:345: in function 'neteval'
        ...rs/snowm/torch/install/share/lua/5.1/nngraph/gmodule.lua:380: in function <...rs/snowm/torch/install/share/lua/5.1/nngraph/gmodule.lua:300>
        [C]: in function 'xpcall'
        /Users/snowm/torch/install/share/lua/5.1/nn/Container.lua:63: in function 'rethrowErrors'
        /Users/snowm/torch/install/share/lua/5.1/nn/Sequential.lua:44: in function 'forward'
        ./densecap/DenseCapModel.lua:253: in function 'forward'
        ./densecap/DenseCapModel.lua:321: in function 'forward_test'
        run_model.lua:77: in function 'run_image'
        run_model.lua:164: in main chunk
        [C]: in function 'dofile'
        ...nowm/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:145: in main chunk
        [C]: at 0x0101c8dd10