tensorflow

colab에 tensorflow 2.x 대 적용하기

gt1000 2019. 11. 13. 21:15

tensorflow 2.0이 출시 된지 한참 지났지만....

여전히 colab 은 1.x 대를 사용하고 있다.

매번

!pip install -q tensorflow==2.0.0

를 실행해서 다운 받고 재시작 하니 시간도 많이 걸리고 귀찮다.

그래서 찾아 봤는데.... 마땅한 방법을 모르겠다.

 

colab에서 아래 코드를 작성하고

# Restart runtime using 'Runtime' -> 'Restart runtime...'

%tensorflow_version 2.x

상단의 런타임 > 런다임 다시 실행

을 해 줘야 2.x 로 반영된다.

 

현재로서는 이게 가장 쉬운거 같다.