Tensorfow填坑记(不定期更新)

版权声明:作者水平有限,博客中难免不少纰漏甚至严重错误,希望大家指正。同时撰写最大的目的也在于交流学习,而不在关注和传播。任重而道远,与您共勉。 https://blog.csdn.net/yexiaohhjk/article/details/89195457

写法类

报错类

could not create cudnn handle: CUDNN_STATUS_NOT_INITIALIZED

参考博客
tensorflow 框架下设置GPU按需分配:

import tensorflow as tf
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
with tf.Session(config=config) as sess:

keras框架(Tensorflow backend) 设置GPU按需分配:

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

猜你喜欢

转载自blog.csdn.net/yexiaohhjk/article/details/89195457