search

xcyan / nips16_PTN

Torch Implementation of NIPS'16 paper: Perspective Transformer Nets
MIT License
141 stars 31 forks source link

Out of memory #7

Closed Santubasa closed 7 years ago

Santubasa commented 7 years ago

Hi, I was just evaluating the pretrained model and encountered the following issue regarding the memory space of the gpu, THCudaCheck FAIL file=/tmp/luarocks_cutorch-scm-1-1355/cutorch/lib/THC/generic/THCStorage.cu line=66 error=2 : out of memory /home/sbasavaraju/torch/install/bin/luajit: ...e/sbasavaraju/torch/install/share/lua/5.1/torch/File.lua:351: cuda runtime error (2) : out of memory at /tmp/luarocks_cutorch-scm-1-1355/cutorch/lib/THC/generic/THCStorage.cu:66 stack traceback: [C]: in function 'read' ...e/sbasavaraju/torch/install/share/lua/5.1/torch/File.lua:351: in function <...e/sbasavaraju/torch/install/share/lua/5.1/torch/File.lua:245> [C]: in function 'read' ...e/sbasavaraju/torch/install/share/lua/5.1/torch/File.lua:351: in function 'readObject' ...e/sbasavaraju/torch/install/share/lua/5.1/torch/File.lua:369: in function 'readObject' /home/sbasavaraju/torch/install/share/lua/5.1/nn/Module.lua:192: in function 'read' ...e/sbasavaraju/torch/install/share/lua/5.1/torch/File.lua:351: in function 'readObject' ...e/sbasavaraju/torch/install/share/lua/5.1/torch/File.lua:369: in function 'readObject' ...e/sbasavaraju/torch/install/share/lua/5.1/torch/File.lua:369: in function 'readObject' /home/sbasavaraju/torch/install/share/lua/5.1/nn/Module.lua:192: in function 'read' ...e/sbasavaraju/torch/install/share/lua/5.1/torch/File.lua:351: in function 'readObject' ...e/sbasavaraju/torch/install/share/lua/5.1/torch/File.lua:369: in function 'readObject' ...e/sbasavaraju/torch/install/share/lua/5.1/torch/File.lua:409: in function 'load' scripts/eval_quant_test.lua:63: in main chunk [C]: in function 'dofile' ...raju/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:150: in main chunk [C]: at 0x00406670

I have Ubuntu 14,04 with graphics model NVIDIA GeForce GTX 470 1 GB , is the graphics not sufficient to run the program ?

Thanks

xcyan commented 7 years ago

No you need to have at least 8G card. You can try to run the code with smaller batchsize and learning rate. But I cannot guarantee the performance. On Tue, Sep 12, 2017 at 6:04 AM Santubasa notifications@github.com wrote:

Hi, I was just evaluating the pretrained model and encountered the following issue regarding the memory space of the gpu, THCudaCheck FAIL file=/tmp/luarocks_cutorch-scm-1-1355/cutorch/lib/THC/generic/THCStorage.cu line=66 error=2 : out of memory /home/sbasavaraju/torch/install/bin/luajit: ...e/sbasavaraju/torch/install/share/lua/5.1/torch/File.lua:351: cuda runtime error (2) : out of memory at /tmp/luarocks_cutorch-scm-1-1355/cutorch/lib/THC/generic/THCStorage.cu:66 stack traceback: [C]: in function 'read' ...e/sbasavaraju/torch/install/share/lua/5.1/torch/File.lua:351: in function <...e/sbasavaraju/torch/install/share/lua/5.1/torch/File.lua:245> [C]: in function 'read' ...e/sbasavaraju/torch/install/share/lua/5.1/torch/File.lua:351: in function 'readObject' ...e/sbasavaraju/torch/install/share/lua/5.1/torch/File.lua:369: in function 'readObject' /home/sbasavaraju/torch/install/share/lua/5.1/nn/Module.lua:192: in function 'read' ...e/sbasavaraju/torch/install/share/lua/5.1/torch/File.lua:351: in function 'readObject' ...e/sbasavaraju/torch/install/share/lua/5.1/torch/File.lua:369: in function 'readObject' ...e/sbasavaraju/torch/install/share/lua/5.1/torch/File.lua:369: in function 'readObject' /home/sbasavaraju/torch/install/share/lua/5.1/nn/Module.lua:192: in function 'read' ...e/sbasavaraju/torch/install/share/lua/5.1/torch/File.lua:351: in function 'readObject' ...e/sbasavaraju/torch/install/share/lua/5.1/torch/File.lua:369: in function 'readObject' ...e/sbasavaraju/torch/install/share/lua/5.1/torch/File.lua:409: in function 'load' scripts/eval_quant_test.lua:63: in main chunk [C]: in function 'dofile' ...raju/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:150: in main chunk [C]: at 0x00406670

I have Ubuntu 14,04 with graphics model NVIDIA GeForce GTX 470 1 GB , is the graphics not sufficient to run the program ?

Thanks

— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub https://github.com/xcyan/nips16_PTN/issues/7, or mute the thread https://github.com/notifications/unsubscribe-auth/AIWuKwLjwZPdm_MW2TeLT5Es4D95eS-Cks5shoFegaJpZM4PUnKJ .