CUDA和CUDNN的坑

一直報這個錯,網上的博客答案也各種各樣。但最簡單的方法 是在caffe的訓練文件中的每個卷積層中加enjine:CAFFE;即不適用cudn的計算引擎,使用caffe自帶的,但是對於我非常深的網絡 一個個手寫添加簡直崩潰,想想還是硬着頭皮重新配環境吧,人生的坑得自己走一邊啊.... https://blog.csdn.net/pkuhyz/article/details/82427384 首先顯卡
相關文章
相關標籤/搜索