[Tensorflow] 텐서플로우2에서 텐서플로우1 사용하기

Sireal·2021년 10월 14일
0

AI

목록 보기
6/12
post-thumbnail

문제

  • import tensorflow as tf
    • 보통 이렇게하면 해당 텐서 버전을 사용할 수 있게 된다.
    • 하지만 텐서플로우1 전용 코드에서 버전2를 쓰면 충돌이 난다.

해결

  • import tensorflow as tf 를 아래의 코드로 대체하자
from tensorflow.python.framework.versions import VERSION
if VERSION >= "2.0.0a0":
    import tensorflow.compat.v1 as tf
    tf.disable_v2_behavior
else:
    import tensorflow as tf
  • 텐서버전이 2이면 1로 바꿔준다.
profile
🚄계속 앞으로🚄

0개의 댓글