将 tf.Session()改成一下形式即可:
...
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
with tf.Session(config=config) as sess:
sess.run(...)
...
Failed to get convolution algorithm. This is probably because cuDNN有效解决
最新推荐文章于 2022-04-13 18:30:23 发布