Skip to content

out of memory on demo_coco #33

Description

@olivain

Hello there,
We tried to test your model on a computer with following configuration :

Ubuntu 16.04 with CUDA 8, CuDNN 5 & Torch 7

The demo_coco.sh failed with following error

THCudaCheck FAIL file=/home/sku/torch/extra/cutorch/lib/THC/generic/THCStorage.cu line=66 error=2 : out of memory /home/sku/torch/install/bin/luajit: /home/sku/torch/install/share/lua/5.1/torch/File.lua:351: cuda runtime error (2) : out of memory at /home/sku/torch/extra/cutorch/lib/THC/generic/THCStorage.cu:66 stack traceback: [C]: in function 'read' /home/sku/torch/install/share/lua/5.1/torch/File.lua:351: in function </home/sku/torch/install/share/lua/5.1/torch/File.lua:245> [C]: in function 'read' /home/sku/torch/install/share/lua/5.1/torch/File.lua:351: in function 'readObject' /home/sku/torch/install/share/lua/5.1/torch/File.lua:369: in function 'readObject' /home/sku/torch/install/share/lua/5.1/nn/Module.lua:192: in function 'read' /home/sku/torch/install/share/lua/5.1/torch/File.lua:351: in function 'readObject' /home/sku/torch/install/share/lua/5.1/torch/File.lua:369: in function 'readObject' /home/sku/torch/install/share/lua/5.1/torch/File.lua:369: in function 'readObject' /home/sku/torch/install/share/lua/5.1/nn/Module.lua:192: in function 'read' /home/sku/torch/install/share/lua/5.1/torch/File.lua:351: in function 'readObject' /home/sku/torch/install/share/lua/5.1/torch/File.lua:409: in function 'load' txt2img_demo.lua:43: in main chunk [C]: in function 'dofile' ...sku/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:150: in main chunk [C]: at 0x00405d50

Error remain the same when changing batch_size from 16 to 8 or 1 in txt2img_demo.lua
Does anyone have any idea to solve this problem ?

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions