添加这段代码即可
import tensorflow as tf
config = tf.compat.v1.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.compat.v1.InteractiveSession(config=config)
出现Failed to get convolution algorithm的解决方法
最新推荐文章于 2024-09-03 17:22:28 发布
添加这段代码即可
import tensorflow as tf
config = tf.compat.v1.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.compat.v1.InteractiveSession(config=config)