Facenet: CUDA_ERROR_OUT_OF_MEMORY

Created on 28 Jul 2017  路  3Comments  路  Source: davidsandberg/facenet

@davidsandberg Even i have set the batch_size to 1, also occur this error
I tensorflow/core/common_runtime/gpu/gpu_device.cc:1030] Creating TensorFlow device (/gpu:0) -> (device: 0, name: GeForce GTX 1060 6GB, pci bus id: 0000:01:00.0)
E tensorflow/stream_executor/cuda/cuda_driver.cc:924] failed to allocate 5.93G (6365773824 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY

Most helpful comment

Try to change GPU memory fraction yo 0.9 or 0.8

All 3 comments

Try to change GPU memory fraction yo 0.9 or 0.8

@pavankumarkatakam thanks

@pavankumarkatakam Great! Thanks!

Was this page helpful?
0 / 5 - 0 ratings

Related issues

Zumbalamambo picture Zumbalamambo  路  3Comments

ouyangbei picture ouyangbei  路  4Comments

haochange picture haochange  路  3Comments

mayank26saxena picture mayank26saxena  路  4Comments

patienceFromZhou picture patienceFromZhou  路  3Comments