본문 바로가기

CS+/설치

install tensorflow 1.15 in colab (20200402 리뉴얼)

( 2020.04.02 기준)

 

아직 GPU가 없는 저 같은분들은 AWS spot 이나 Colab 을 사용해서 예제들을 돌릴텐데요.

 

이제 코랩에서 tensorflow 의 디폴트는 2.x 버전이 되었습니다. 

 

1) tensorflow 버전 확인 in colab

 

import tensorflow as tf
print(tf.__version__)

이제 코랩에서  tensorflow 는 2점대 이므로 만약 1점대를 원하신다면 제거하고 설치해주셔야 합니다. 

 

 

2) Uninstall tensorflow==2.x in colab

 

!pip uninstall tensorflow
!pip install tensorflow==1.15

#런타임 재실행! 

import tensorflow as tf
print(tf.__version__)

 

 

먼저 삭제해주세요. 

명령어 실행하면 계속진행할거냐고 묻습니다. y 눌러주시면 됩니다 :)

그리고 설치하면 됩니다! 

 

런타임 꼭 다시 실행해주세요!

 

런타임을 다시 실행해야 1.15 로 실행이 됩니다! 

tensorflow 1.15 Import 되는 걸 볼 수 있습니다 :) 

 

 

+) % 활용한 텐서플로우 설치 

 

%tensorflow_version 2.x 

 

를 입력한 뒤에 텐서플로우를 Import 하면 최신 버전이 불려집니다.(위 캡쳐는 1월초에 한거에요) 아마 위의 매직명령어는 최신 버젼으로 무조건 텐서플로우를 불러주는 것 같습니다. 왜 그러냐면

 

위처럼 특정버젼을 입력하면 선택사항이 1.x 아니면 2.x 이렇게 밖에 없다고 하네요. 

 

여튼 위와 같은 매직명령어도 있으니 참고하시면 좋을 거 같습니다!

 

이상 코랩에서 2점대 텐서플로우 제거하고 텐서플로우 1.15 설치하기였습니다 :)