当时用Gpu进行训练时,我遇到了如下错误
Failed to get convolution algorithm. This is probably because cuDNN failed to initialize,
网上搜索了很多方法都没有解决,直到昨天看到一篇博客,在添加一行代码,实测可以解决问题。
代码如下:
config = tf.compat.v1.ConfigProto(gpu_options=tf.compat.v1.GPUOptions(allow_growth=True))
sess = tf.compat.v1.Session(config=config)
GPU训练错误
最新推荐文章于 2023-04-03 18:56:57 发布