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.
반응형