Colab에서 Tensorflow 1.X 버전 사용하기, 코랩 텐서플로우 버전 낮추기

텐서플로우가 새롭게 버전을 업데이트되면서 Tensorflow 2 가 만들어졌습니다. 텐서플로우가 1.X에서 2.X로 변하면서 다양한 문법적, 함수적 변화가 발생했습니다. 이에 따라서, 이전에 만들었던 텐서플로우 1의 소스코드를 2에서 사용하기 어려워졌습니다. 없어진 함수들도 있고 등등등...ㅠ

이번 포스팅을 통해서 구글 코랩에서 텐서플로우 버전을 낮춰 Tensorflow1.14 또는 1.15 등의 1 버전을 사용하기 위한 방법을 알아보겠습니다.

1. Colab Tensorflow version 확인

import tensorflow as tf

tf.__version__

>> 2.2.0

먼저 텐서플로우 버전을 확인하도록 합니다. 사실 안 해도 무방합니다...ㅎ

당연히 기본적으로 colab의 텐서플로우는 2.2.0 또는 그 이상의 버전이 기본으로 세팅되어있습니다.

 

2. Colab Tensorflow 2 삭제

!pip uninstall tensorflow

현재 코랩에 설치된 텐서플로우 2 버전을 제거해줍니다.

 

잊지말고 y 값 넣어주세요. 바보같은 저 같이 아무생각없이 기다리면 계속 기다리기만합니다...ㅠ

 

3. Colab Tensorflow 1.X 버전 설치

!pip install tensorflow==(원하는 텐서플로우 버전)

!pip install tensorflow==1.15 # 1.15 버전 Tensorflow 설치

!pip install tensorflow==1.14 # 1.14 버전 Tensorflow 설치

tensorflow를 pip명령어를 사용해서 설치합니다.  단, 꼭 버전을 명시해주셔야합니다.

 

저는 텐서플로우 1.15 버전을 설치하였습니다. 원하는 버전을 지정해서 설치해주세요.

 

4. Colab 런타임을 재실행

설치가 완료된 이후에 발생되는 warning이 발생합니다. 설치가 완료된 이후에는 런타임을 다시 실행해야 재설치된 1.x 버전의 Tensorflow를 사용할 수 있습니다. 

 

5. 주의할 점

주의할 점 ! 

1. 코랩 연결 후, 위의 작업을 바로 진행해주세요. mount하거나 directory 변경을 하지 않은 상태로 진행하시는 것이 좋습니다.

2. '런타임 다시 실행하기'를 꼭 해야 바뀐 버전을 사용할 수 있습니다. 런타임 중지 or 다시시작이 아닙니다.