faustomilletari / VNet

GNU General Public License v3.0
284 stars 123 forks source link

I meet a problem about out of memory #27

Open wangyidada opened 7 years ago

wangyidada commented 7 years ago

when I run the main.py , it says Check failed: error == cudaSuccess(2 vs. 0) out of memory. I use Nvida Titan X with 12G of memory . Who can help me to solve this problem?

faustomilletari commented 7 years ago

if you look at the CLOSED issues you will find an answer to your question :)

(It’s about cudnn)

On Apr 12, 2017, at 3:24 AM, wangyidada notifications@github.com wrote:

when I run the main.py , it says Check failed: error == cudaSuccess(2 vs. 0) out of memory. I use Nvida Titan X with 12G of memory . Who can help me to solve this problem?

— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub https://github.com/faustomilletari/VNet/issues/27, or mute the thread https://github.com/notifications/unsubscribe-auth/AMtsvnS_7EH7oFhkjS19AoysJwJ3Re7bks5rvHwZgaJpZM4M7C8C.

sagarhukkire commented 7 years ago

@wangyidada

its not Vnet issue , check your gpu usage

run command: nvidia-smi

if you are utilizing limited space GPU then reduce input data

mren888 commented 7 years ago

Make sure you use 3D-Caffe and CuDNN V5.