search

liuzhuang13 / DenseNet

Densely Connected Convolutional Networks, In CVPR 2017 (Best Paper Award).
BSD 3-Clause "New" or "Revised" License
4.73k stars 1.07k forks source link

Cuda out of memory when use memory efficient densenet for deploying #44

Open JannaShen opened 6 years ago

JannaShen commented 6 years ago

WARNING: Logging before InitGoogleLogging() is written to STDERR F0129 15:24:34.494936 153936 DenseBlock_layer.cu:203] Check failed: error == cudaSuccess (2 vs. 0) out of memory Check failure stack trace: WARNING: Logging before InitGoogleLogging() is written to STDERR F0129 18:17:47.501026 32543 syncedmem.cpp:71] Check failed: error == cudaSuccess (2 vs. 0) out of memory Check failure stack trace:

When i deploy the trained model with memory efficient densenet with matlab, i always counter this problem . I think it will be solved to reset cuda memory after each time calculation, can you give me some advice to solve this problem.

JannaShen commented 6 years ago

so for the caffe version of efficient densenet, need to delete the named Cuda memory after each deployment .