Unknown: Failed to get convolution algorithm.

  1. 一般可能是因爲cudnn的版本沒有安裝對,先用下邊命令來檢測一下,看一下你cudnn的版本對應的對不對? 
    cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
    #此命令適用於Ubuntu系統,如果是Windows,那麼百度搜一下怎麼檢查cudnn版本

  2.        你的顯存暴炸了,也就是GPU沒有分配好,在你使用的代碼中添加下邊的代碼就好了。       
    gpus = tf.config.experimental.list_physical_devices(device_type='GPU')
    tf.config.experimental.set_memory_growth(gpus[0],True)
                                                                                                                                                                         
  3.  看看你的代碼是不是有問題,tensorflow1.0寫的代碼,不能直接在tensorflow2.0以上版本運行
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章