LSTM 레이어 사용시 cuDNN 관련 에러 나는 경우 해결방법Deep Learning & Machine Learning/강좌&예제 코드2021. 9. 9. 20:22
Table of Contents
반응형
구글에서 검색해봐도 해결방법이 보이지 않았던 에러입니다. 원인이 여러가지 일 수 있겠지만 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.
반응형
'Deep Learning & Machine Learning > 강좌&예제 코드' 카테고리의 다른 글
Tensorflow 디버깅 정보 메시지 안보이게 하기 (0) | 2021.09.14 |
---|---|
Tensorflow Dataset의 window 메소드 사용법 (0) | 2021.09.14 |
Unknown: OSError: cannot identify image file (0) | 2021.08.21 |
손글씨 숫자 인식하여 세븐 세그먼트에 출력하기 (0) | 2021.06.16 |
텐서플로우 강좌 - 텍스트 분류(text classification) (0) | 2021.05.09 |