TensorFlow中出现 CUDA_ERROR_OUT_OF_MEMORY 的错误解决方案

我是在用谷歌开源的facenet工程，使用自带的MTCNN用来align人脸数据集时遇到这个问题的。

这个问题出现的原因在于TF想获得全部的GPU资源，设总资源为M，现在只能为TF分配N，N<M，然后TF就不干了。

对于facenet这个工程来说，只需把arg中的可选参数 gpu-memory-fraction 设为0.7，反正必须小于1，是能N/M的百分比。

 

也可以这么做

在定义session之前，增加
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7)
config = tf.ConfigProto(log_device_placement=False,gpu_options=gpu_options)
with tf.Session(config=config)


也可以
config.gpu_options.allow_growth=True

 

参考：

https://blog.csdn.net/tengxing007/article/details/78535640

https://blog.csdn.net/wangkun1340378/article/details/72782593