Deep Learning & Machine Learning/강좌&예제 코드

LSTM 레이어 사용시 cuDNN 관련 에러 나는 경우 해결방법

webnautes 2021. 9. 9. 20:22
반응형

구글에서 검색해봐도 해결방법이 보이지 않았던 에러입니다. 원인이 여러가지 일 수 있겠지만 Keras의 LSTM 레이어에 activation='relu'를 추가해놓았다면 제거해주면 해결됩니다.  

 

 

WARNING:tensorflow:Layer lstm will not use cuDNN kernels since it doesn't meet the criteria. 

It will use a generic GPU kernel as fallback when running on GPU.

반응형