[Tensorflow] 텐서플로우 버전 확인, 업그레이드, 다운그레이드 방법

텐서플로우 버전(version) 확인 및 변경

텐서플로우(Tensorflow)를 사용하시다 보면 버전 호환성 문제로

버전을 업그레이드 혹은 다운그레이드해야 하는 경우가 가끔씩 발생합니다.

 

이번 시간에는 해당 모듈의 버전(version)을 간단하게 확인하는 방법과

업그레이드, 다운그레이드하는 변경 방법에 대해서 살펴보도록 하겠습니다.

 

 

텐서플로우 버전 확인(Tensorflow version check)

Tensorflow 모듈의 버전 확인 방법은 모듈을 import한 뒤,

__version__ 속성 값을 조회하여 쉽게 확인이 가능합니다.

import tensorflow as tf

print(tf.__version__)
# 2.5.0

저는 2.5.0 버전의 텐서플로우를 사용 중이었습니다.

 

version 확인 방법은 gpu 사용 여부에 관계 없이 동일한 듯 합니다.

 

 

텐서플로우 버전 업그레이드, 다운그레이드

텐서플로우 버전을 업그레이드, 다운그레이드하는 방법은 동일합니다.

 

gpu 사용 여부에 따라 명령어가 약간 다른데,

터미널 창 혹은 ipynb 파일의 셀 내에 아래의 명령어를 작성해주시면 됩니다.

# gpu 버전이 아닌 경우
pip install --upgrade tensorflow==버전
# gpu 버전인 경우
pip install --upgrade tensorflow-gpu==버전

# 업그레이드 예시
pip install --upgrade tensorflow==2.7.0
# 다운그레이드 예시
pip install --upgrade tensorflow==1.15.0

위 방법으로 1.x 버전으로도 다운그레이드가 가능했습니다.