컴공

Tensorflow1버전기능 Tensorflow2에서 사용하기

비온뒤엔습해 2022. 6. 27. 14:30

최근 공부하고 있는 tensorflow의 강의코드가 tensorflow1버전 기반이여서 골머리를 앓던 중

여러 인터넷 검색을 기반으로 다음과 같은 코드를 찾았다.

 

내 환경

:Pycharm 

 Python 3.9

 Tensorflow 2.9.1

 많은 분들이

import tensorflow.compat.v1 as tf

tf.disable_v2_behavior()

를 알려주시던데 내 경우에는 compat이라는 module이 존재하지 않다고 뜨면서 되지 않았다.(no module named 'compat')

검색하다가 다음 사이트를 발견했다.

https://youtrack.jetbrains.com/issue/PY-43918

 

Unresolved reference for tensorflow.compat, tensorflow.data : PY-43918

Old summary: PyCharm Inspection claims no modules named …. , and Code Completion fails although the name is valid and the code runs Python 3.8 via homebrew, tensorflow 2.3.0 installed with pip import tensorflow as tf import tensorflow.compat.v1 as v1 Pyc

youtrack.jetbrains.com

해당 링크의 댓글에서 import tensorflow._api.v2.compat.v1 as v1를 제시해줬고 

 

파일의 앞부분에 import tensorflow._api.v2.compat.v1 as v1

                           tf.disable_v2_behavior()

코드를 넣어줬더니 정상적으로 작동했다.