tensorflow显存不足报错CUBLAS_STATUS_ALLOC_FAILED解决

tensorflow显存不足报错CUBLAS_STATUS_ALLOC_FAILED解决

TensorFlow执行，报显存不足错误，如下

解决方法

通过设定config为使用的显存按需自动增长，避免显存被耗尽，可进行有效的预防显存不足问题。

# 定义TensorFlow配置
config = tf.ConfigProto()
# 配置GPU内存分配方式，按需增长，很关键
config.gpu_options.allow_growth = True
# 配置可使用的显存比例
config.gpu_options.per_process_gpu_memory_fraction = 0.2
# 在创建session的时候把config作为参数传进去
sess = tf.InteractiveSession(config = config)

官网：https://tensorflow.google.cn/guide/using_gpu
简书 https://www.jianshu.com/p/0d9491e84e1c
CSDN博客 https://blog.csdn.net/feixiang7701/article/details/81515447