faustomilletari/VNet

I meet a problem about out of memory

wangyidada opened this issue · 3 comments

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?

@wangyidada

its not Vnet issue , check your gpu usage

run command: nvidia-smi

if you are utilizing limited space GPU then reduce input data

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