bearsprogrammer / real-time-deep-face-recognition

using facenet algorithm
MIT License
233 stars 100 forks source link

How much graphics memory for mtcnn model in the face detection #10

Open so-as opened 7 years ago

so-as commented 7 years ago

hi, bearsprogrammer! when I run this project to do face detect with tensorflow 1.2 version on the ubuntu 14.04, I met the error: 2017-10-13 00:39:02.833148: I tensorflow/core/common_runtime/gpu/gpu_device.cc:965] Found device 0 with properties: name: Tesla M40 major: 5 minor: 2 memoryClockRate(GHz): 1.112 pciBusID: 0000:82:00.0 totalMemory: 11.93GiB freeMemory: 11.82GiB 2017-10-13 00:39:02.833193: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1055] Creating TensorFlow device (/device:GPU:0) -> (device: 0, name: Tesla M40, pci bus id: 0000:82:00.0, compute capability: 5.2) 2017-10-13 00:39:02.863038: E tensorflow/stream_executor/cuda/cuda_driver.cc:936] failed to allocate 11.93G (12808486912 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY It seem the 12G memory is not enough for mtcnn model. Could you tell me which GPU device you use, how much memory does the mtcnn detection algorith need? thank you very much, wait for your reply! @bearsprogrammer