美文网首页
tensorflow yolov3 Could not crea

tensorflow yolov3 Could not crea

作者: bluekhoja | 来源:发表于2019-11-15 22:42 被阅读0次

环境说明

python 3.7

tensorflow 1.5.0

cuda 10.0

cudnn 7.5.0

解决方法

tensorflow框架下设置GPU按需分配:

config = tf.ConfigProto()

config.gpu_options.allow_growth =True

with tf.Session(config=config)as sess:

keras下设置GPU按需分配:

config = tf.ConfigProto()

config.gpu_options.allow_growth = True

sess = tf.Session(config=config)

K.set_session(sess)

相关文章

网友评论

      本文标题:tensorflow yolov3 Could not crea

      本文链接:https://www.haomeiwen.com/subject/moleictx.html