服務器環境
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
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