search

twni2016 / OrganSegRSTN_PyTorch

PyTorch implementation of OrganSegRSTN - CVPR 2018
http://ccvl.jhu.edu/
MIT License
108 stars 33 forks source link

My problem #9

Open liuwenhaha opened 4 years ago

liuwenhaha commented 4 years ago

when I run this 'run.sh' on my GPU it will show 'cuda runtime error(2): out of memory ' on my CPU it will show '$ Torch: not enough memory: you tried to allocate 0GB'

twni2016 commented 4 years ago

Is your GPU memory larger than 10G? We cannot run on CPU.

liuwenhaha commented 4 years ago

Is your GPU memory larger than 10G? We cannot run on CPU.

Yes, I'm sure my GPU has 32gb of memory, and I'm following your run.sh exactly

liuwenhaha commented 4 years ago

Is your GPU memory larger than 10G? We cannot run on CPU.

Since the GPU mode didn't work, I tried the CPU mode, which also didn't work. Also, I tried to read only fewer files, such as three, which also appeared'cuda runtime error(2): out of memory'