- 最近在使用TF2.0,運行程序時出現(xiàn)以下錯誤：
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. [Op:Conv2D]
- 一開始懷疑是CUDA和CuDNN配置錯誤。反復(fù)試驗后發(fā)現(xiàn)可能是GPU內(nèi)存不足造成的。需要在程序前加以下一段代碼：
from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession
config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)
- 意思是對GPU進行按需分配。主要原因是訓(xùn)練圖像較大，消耗GPU資源較多。但我的RTX2060顯存只有6GB，所以會出現(xiàn)這個錯誤。
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成，瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明：文章內(nèi)容（如有圖片或視頻亦包括在內(nèi)）由作者上傳并發(fā)布，文章內(nèi)容僅代表作者本人觀點，簡書系信息發(fā)布平臺，僅提供信息存儲服務(wù)。
