tsinghua-rll / VoxelNet-tensorflow

A 3D object detection system for autonomous driving.
MIT License
453 stars 123 forks source link

ResourceExhaustedError (see above for traceback): #24

Closed ashishkumar-rambhatla closed 6 years ago

ashishkumar-rambhatla commented 6 years ago

Hey Jeasinema!

I'm getting the following error when running on a system with 2gb nvidia geforce 940M card with ompute capability 5.0 ResourceExhaustedError (see above for traceback): OOM when allocating tensor with [shape[1,10,400,352,128]] error can you please tell me if i can run training with my current system configuration. If so, please give me the suitable input arguments with train.py so as to avoid over usage of gpu.

jeasinema commented 6 years ago

Hi, You need at least one nvidia gpu with more than 6GB memory.