jainal09
commented
4 years ago 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
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？