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