프로그래밍/Python

[Python] Tensorflow Epoch 1/10 멈춤 에러 해결

Beginner:) 2021. 9. 1.
320x100

텐서플로우 이미지 분류 예제를 따라하다보면 각종 에러를 만난다 ...

 

2021.09.01 - [프로그래밍] - [Python] Tensorflow cudart64 not found error 해결

 

 

2021.09.01 - [프로그래밍] - [Python] tensorflow cusolver64_11.dll, cudnn64_8.dll not found 해결

 

 

만나다 보면 ... 모델 훈련에서 

 

Epoch 1/10 뒤로 프로그램이 죽는경우가 있다.

 

에러도 안뱉고 ....

 

구글링을 하면 GPU 설정, 텐서플로 버전을 낮추라는 등의 조언이 있지만 

 

앞선 링크중 두번째 것, cudnn64_8.dll 을 추가하는 부분에서

 

해당 라이브러리만 넣지말고 

 

bin 폴더에있던 모든것을 다 넣어버리니 해결됐다.. 

 

애초에 관련된 라이브러리들일텐데 에러만 따라가다보니 놓친 부분이다

 

실제로 나머지 라이브러리도 넣고 실행하면 Loaded cuDNN version - 메시지가 출력된다. 

(사실 8.1을 넣어야하는데 이것저것 해보다 8.2를 넣어서 그럴수도? 있다)

 

 

반응형

댓글