环境:Tensorflow 1.13.1,cuda10, cudnn7.6,python3.7
我使用了两个模型用于识别与追踪,载入第二个模型时,出错了
报错如题
报错的最后一行:
UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
可能是GPU内存不足
解决方法:
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config)
我是这个原因,看别人出现的问题有可能是版本不匹配。