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
结果:
2020-