Error : 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. #24828

pip install --upgrade tensorflow-gpu 명령어를 입력하면 최신 버전의 Tensorflow-gpu가 설치된다(2019년 6월 1일). 현재 Ubuntu16.04에 CUDA 9.0, cuDNN 7.1.4 가 설치되어 있는데, 위의 에러는 Tensorflow-gpu 1.13.1 이 설치되었을 때 발생했었다. 해결 방법은 Tensorflow-gpu를 1.8.0 으로  downgrade하는 것이다.

 

pip install tensorflow-gpu==1.8.0

 

[ubuntu16.04, cuda9.0, cudnn7.1.4]