未知错误:获取卷积算法失败。这可能是因为cuDNN失败了。
这可能是由于显存被占用,详细原因可看这篇。
关于keras or tensorflow 显存占用问题 :
keras or tensorflow 显存占用问题 - 简书
解决方法:
1. 如果用的是Keras,在jupyter notebook 最开始添加代码:
import tensorflow as tf
import numpy as np
import keras
config = tf.compat.v1.ConfigProto(allow_soft_placement=True)
config.gpu_options.per_process_gpu_memory_fraction = 0.3
tf.compat.v1.keras.backend.set_session(tf.compat.v1.Session(config=config))
#放在代码顶部的导入包的位置
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
如果是tensorflow:
from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import Interactive