浅谈神经网络常规异常解决办法（一直更新）

前言

记录下一些小问题的解决方案（长期更新）。

tensorflow和keras限制gpu显存

tensorflow

ps：注意keras和tensorflow设置方式是不同的
限制为0.5

import tensorflow 
tf_config = tensorflow.ConfigProto()  
tf_config.gpu_options.per_process_gpu_memory_fraction = 0.5 # 分配50%  
session = tensorflow.Session(config=tf_config)

自适应最佳

tf_config = tensorflow.ConfigProto()  
tf_config.gpu_options.allow_growth = True  
session = tensorflow.Session(config=tf_config) 

keras

import keras
import tensorflow as tf
from keras.backend.tensorflow_backend import set_session  
config = tf.ConfigProto()  
config.gpu_options.allow_growth = True  
set_session(tf.Session(config=config))  

设置tensorflow和keras设置cpu运行

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"  
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

常见异常解决方案

这种异常我个人认为是tensorflow版本导致

tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
         [[{{node Conv/convolution}}]]
         [[{{node metrics/_f_score/Identity}}]]

解决方法：

import tensorflow as tf
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config)