解决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

你可能感兴趣的:(Tensorflow)