我的代码中,产生这个错误的解决办法是添加
config = tf.ConfigProto(gpu_options=tf.GPUOptions(allow_growth=True))
sess = tf.Session(config=config)
限制显存的增长,不一定适用全部~
博客提到代码中出现错误,解决办法是添加限制显存增长,但此方法不一定适用于所有情况。
我的代码中,产生这个错误的解决办法是添加
config = tf.ConfigProto(gpu_options=tf.GPUOptions(allow_growth=True))
sess = tf.Session(config=config)
限制显存的增长,不一定适用全部~
4670
3299

被折叠的 条评论
为什么被折叠?