报错：tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm.

最新推荐文章于 2023-03-27 12:57:27 发布

燕芝沛然

最新推荐文章于 2023-03-27 12:57:27 发布

阅读量1.2k

点赞数

分类专栏： AI 文章标签： gpu tensorflow

本文链接：https://blog.csdn.net/weixin_44205550/article/details/104018889

版权

UnknownError: Failed to get convolution algorithm.

运行sess.run()
错误行标定这一行：sess.run(optimizer, feed_dict={ x: batch_x, y: batch_y, keep_prob: dropout }
显示错误信息：tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm.

分析原因和解决方案
方法一：IDE运行加载多次代码，tensorflow-cpu 与tensorflow-gpu有相同接口，卷积时，可能调了不匹配的那一个。可以重启再重新运行即可（我是重启后运行正常）

方法二：在代码中添加gpu的配置说明

import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0' #use GPU with ID=0
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.5 # maximun alloc gpu50% of MEM
config.gpu_options.allow_growth = True #allocate dynamically