search

thtrieu / darkflow

Translate darknet to tensorflow. Load trained weights, retrain/fine-tune using tensorflow, export constant graph def to mobile devices
GNU General Public License v3.0
6.14k stars 2.08k forks source link

out_of_memory issue on jetson tx2 #966

Open varun-tangoit opened 5 years ago

varun-tangoit commented 5 years ago

Hey Hi,

im trying to implement this on my jetson tx2, when i try compile i got error out of memory but i dont know, how to resolve this issue with respect to jetson.

W tensorflow/core/common_runtime/bfc_allocator.cc:215] Allocator (GPU_0_bfc) ran out of memory trying to allocate 2.13GiB. The caller indicates that this is not a failure, but may mean that there could be performance gains if more memory were available.
2019-01-08 14:17:28.210777: W tensorflow/core/common_runtime/bfc_allocator.cc:215] Allocator (GPU_0_bfc) ran out of memory trying to allocate 3.54GiB. The caller indicates that this is not a failure, but may mean that there could be performance gains if more memory were available.

But i have around totalMemory: 7.66GiB freeMemory: 5.04GiB

Please help us, to fix this issue and performance also i get around 2-3 fps only

weiwei106 commented 5 years ago

I think you can use low batch size, set --batch_size 1.