Ipython에서 tensorflow version2 사용 자동 적용

cybertramp

2019/09/23

Categories: etc Tags:

Jupyter Notebook의 Ipython에서 tensorflow를 사용할때 최신 버전인 version2가 설치되어 있는 경우가 잘 없다(현 시점 Google colab, Zepl). 따라서 해당 노트북을 매번 실행할때 pip 명령을 통해 매번 기존 tensorflow를 제거하고 최신 버전을 설치하는 작업을 매번 무조건 거치게된다. 문제는 이것이 시간이 조금 걸린다는 것이다. 이로 인한 시간을 조금 줄여보고자 한다.

아래는 %%bash를 활용하여 작성한 코드이다. 이를 상단에 넣고 원하는 작업을 하면된다.

해당 코드는 gist 링크에서 내려 받을 수 있다.

# Tensorflow version check & upgrade
# Written by paran_son@outlook.com
# Updated 19.09.23

%%bash
var=`python3 -c 'import tensorflow as tf; print(tf.__version__)'`
if [ "${var:0:1}" = "2" ];then
  echo "version2"
else
  echo "version1"
  pip uninstall tensorflow -y
  pip uninstall tensorflow-gpu -y
  pip install tensorflow-gpu==2.0.0-beta1
fi

소스코드를 간단히 살펴보면 %%bash 로 bash 블록을 만들고

python3 -c 'import tensorflow as tf; print(tf.__version__)'로 버전을 var에 넣고

"${var:0:1}" = "2" 조건을 통해 var의 값의 첫자가 "2"인지 확인하여 검사하게 된다.

else 인 경우 기존 version 1의 tensorflow를 제거 하고 tensorflow 2.0 버전을 설치하게 된다.

comments powered by Disqus