原文:神經網絡訓練時出現 無法獲取卷積算法的問題 tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize

使用訓練環境: Tensorflow . . ,CUDA . ,cuDNN . . 。 當實例化VGG 卷積基時,出現了無法獲取卷積算法的問題,提示可能因為cuDNN無法初始化。 打開代碼文件在前面添加如下行,問題得以解決。 ...

2020-08-09 16:47 1 2460 推薦指數:

查看詳情

Failed to get convolution algorithm解決

蒸騰了兩天,終於搞定了 是cudnn版本的問題 更新cudnn的時候,首先要刪除/usr/local/cuda-10.0/targets/x86_64-linux/lib路徑下所有之前cudnn版本的庫,sudo rm libcudnn* 然后 將下載的新版本的cudnn拷貝到 ...

Wed Feb 26 00:32:00 CST 2020 0 1066
 
粵ICP備18138465號   © 2018-2026 CODEPRJ.COM