環境: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)
我是這個原因,看別人出現的問題有可能是版本不匹配。