jcjohnson/densecap

out of memory

lostnighter opened this issue · 1 comments

THCudaCheck FAIL file=/tmp/luarocks_cutorch-scm-1-7860/cutorch/lib/THC/generic/THCStorage.cu line=66 error=2 : out of memory
/usr/local/bin/luajit: /usr/local/share/lua/5.1/nn/utils.lua:11: cuda runtime error (2) : out of memory at /tmp/luarocks_cutorch-scm-1-7860/cutorch/lib/THC/generic/THCStorage.cu:66
stack traceback:
[C]: in function 'resize'
/usr/local/share/lua/5.1/nn/utils.lua:11: in function 'torch_Storage_type'
/usr/local/share/lua/5.1/nn/utils.lua:57: in function 'recursiveType'
/usr/local/share/lua/5.1/nn/Module.lua:160: in function 'type'
/usr/local/share/lua/5.1/nn/utils.lua:45: in function 'recursiveType'
/usr/local/share/lua/5.1/nn/utils.lua:41: in function 'recursiveType'
/usr/local/share/lua/5.1/nn/Module.lua:160: in function 'type'
/usr/local/share/lua/5.1/nn/utils.lua:45: in function 'recursiveType'
/usr/local/share/lua/5.1/nn/utils.lua:41: in function 'recursiveType'
/usr/local/share/lua/5.1/nn/Module.lua:160: in function 'type'
/usr/local/share/lua/5.1/nn/utils.lua:45: in function 'recursiveType'
/usr/local/share/lua/5.1/nn/utils.lua:41: in function 'recursiveType'
/usr/local/share/lua/5.1/nn/Module.lua:160: in function 'type'
/usr/local/share/lua/5.1/nn/utils.lua:45: in function 'recursiveType'
/usr/local/share/lua/5.1/nn/Module.lua:160: in function 'type'
./densecap/DenseCapModel.lua:199: in function 'convert'
extract_features.lua:65: in function 'main'
extract_features.lua:100: in main chunk
[C]: in function 'dofile'
/usr/local/lib/luarocks/rocks/trepl/scm-1/bin/th:150: in main chunk
[C]: at 0x00406200

My GPU memory has 12206MiB. Any idea?

Hey,
Check this docker image I created for running dense cap in gpu!
https://hub.docker.com/r/jainal09/densecap_gpu

It runs perfectly in it.

For Cpu version.
Use this docker file.
https://github.com/runwayml/alpha_models/blob/master/densecap/Dockerfile