본문 바로가기
CS/삽질로그

[tensorflow] Graph execution error

by judy@ 2023. 4. 26.

아니...이렇게 직관적이지 않은 에러는 마주할 때마다 당황스럽다. 하지만 해결 완! 당황해서 캡쳐는 못했지만..

 

  • 문제 상황: 모델을 쌓고(커스텀), 학습을 위해 fit을 했는데 Graph execution error 발생
  • 해결 방법: GPT, 구글링을 통해 CUDA, tensorflow 버전의 매칭 오류가 추측됨. 확인해보니 현재 CUDA 11.6 버전인데, tensorflow 2.12가 깔려있었음. 아래 글에 첨부된 링크를 통해서 cuda와 tensorflow 버전을 확인했고 11.6에 대해서는 명시되어 있지는 않았지만 2.11로 하면 될 것 같은 느낌적인 느낌. 아래와 같이 버전을 다운그레이드하니 해결됨
# uninstall tensorflow 2.12
$ pip uninstall tensorflow

# reinstall tensorflow 2.11
$ pip install tensorflow==2.11

https://data-scient2st.tistory.com/81

 

[tensorflow] tensorflow CUDA 호환성 확인하기

다음 링크에서 확인할 수 있음 https://www.tensorflow.org/install/source#tested_build_configurations 소스에서 빌드 | TensorFlow Check out sessions from the WiML Symposium covering diffusion models with KerasCV, on-device ML, and more. Watch

data-scient2st.tistory.com

 

반응형