原文:Failed to get convolution algorithm. This is probably because cuDNN failed to initialize

在訓練的腳本開頭添加以下代碼,指定使用的GPU,然后進行動態分配內存,重新運行就可以。 當然,出現這種問題也存在cudnn版本問題,但是發現這種情況並不多,大多是上面這種。 ...

2020-05-14 08:08 0 2782 推薦指數:

查看詳情

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-2025 CODEPRJ.COM