Out of memory issue during training and testing
ZezhouCheng opened this issue · 1 comments
ZezhouCheng commented
My env setting is Pytorch 0.4.0 + CUDA 9.0, I disabled the CUDNN batch norm. However, it seems that there is a GPU memory leak. Does anyone encounter this problem?
ZezhouCheng commented
Problem solved after switching to another server ...