[딥러닝] CNN 학습 불가능 에러

[김현우]

딥러닝 과정 수강중입니다.

Project 1까지는 아무 문제 없이 학습이 성공했는데, Project 2에서 CNN 학습을 시킬 때 콘솔창에 아래와 같은 콘솔 에러가 뜨고 학습이 진행되지 않습니다.

 

 

Convolution Layer와 Pooling Layer가 없을 때는 정상 작동하고, 끼워넣으면 저런 에러가 뜨는 것 같습니다. 구글링 해봐도 원인을 찾기 어렵네요.

PyCharm 환경에서 개발중이고, GPU는 RTX 3060 Laptop이고, Tensorflow, GPU 드라이버, CUDA Toolkit, cuDNN은 전부 최신 버전으로 세팅해놨습니다.

혹시 몰라서 텐서플로우 버전을 낮춰서 재설치해도 여전히 같은 에러가 뜨네요. 현 상황에서 추정되는 원인이 있으면 알려주시면 감사하겠습니다.

[codingapple]

google colab에서는 정상적으로 되면

gpu 관련해서 설치하는 것들 버전이 잘못되었을 수도 있습니다

항상최신버전이 아니라 https://www.tensorflow.org/install/source_windows#gpu 여기서 버전을 다 맞춰야 잘될걸요 

[글쓴이]

글