failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED

環境: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)

我是這個原因,看別人出現的問題有可能是版本不匹配。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章