解決Failed to get convolution algorithm. cudnn failed to initialize...

CuDNN library compatibility error after loading model weights

服務器環境

Ubuntu16.04、cuda9.0、cudnn7.0、tensorflow1.12.0、python3.5、1080ti

錯誤

Failed to get convolution algorithm. cudnn failed to initialize…

原因

tensorflow版本太高

解決方案

sudo pip3 install --upgrade --force-reinstall tensorflow-gpu==1.9.0 --user

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章