본문 바로가기
  • SDXL 1.0 + 한복 LoRA
  • SDXL 1.0 + 한복 LoRA
Study/인공지능학습

TensorFlow 1.0.0 설치 - Docker에서...

by 마즈다 2017. 2. 23.
반응형


Docker에서 TensorFlow 1.0 업그레이드 하기

최근 텐서플로우가 1.0을 발표하면서 많은 분들이 1.0 업그레이드에 여념이 없다. 이런 시류에 편승해서 나도
Docker 이미지로 설치한 텐서플로우 0.12.1 버전을 1.0.0 버전으로 업그레이드를 해보기로 했다.


사실 뭐가 어떻게 변했는지도 모르고 그냥 남들 하니까 따라한다…ㅠ.ㅠ
게다가 Docker 자체를 잘 모르는 상태에서 달리 뭘 어찌해야 하는지도 잘 모르겠고. 텐서플로우 홈페이지에 가도
그냥 기존 코드를 1.0에 맞게 변경하는 내용만 있고…Installing 항목에는 기존과 달라진 내용은 안보이고…
가장 상투적인 것이 가장 확실하다고 했던가. 그냥 기존 버전 삭제 후 재설치 하기로 했다.


그 과정에서도 많은 시행착오를 거쳤지만 굳이 그 모든 실수를 다 거론할 필요는 없을 것 같아 깔끔한 성공 과정만
정리한다.



텐서플로우 버전 확인

우선 업그레이드를 하고 나서 제대로 업그레이드가 되었는지 확인이 필요하니 먼저 텐서플로우 버전을 확인하는
방법을 확인해보았다. 간단하게 아래 그림과 같이 코드를 입력하면 된다. 이 화면은 jupyter Notebook 화면이다.



오호라 0.12.1 버전이 현재 내가 사용하는 버전이구나. 업그레이드 해야지~


Docker에 설치된 텐서플로우 이미지 확인

콘솔 창에서 아래 이미지와 같이 명령어를 입력하면 현재 설치된 모든 이미지의 목록이 나온다. 나는 이번에 처음
텐서플로우를 설치하면서 Docker를 사용했기에 이미지가 달랑 하나다. 만약 많은 이미지가 있다면 아래의 코드와
같이 필터링 할 수 있다.

docker images | grep tensorflow




Docker에 설치된 이미지 삭제

역시 아래 코드와 같이 입력하면 된다. 

docker rmi [이미지 ID]

그런데…뭔가 에러가 떴다. 정지된 컨테이너에서 이 이미지를 사용하고 있단다. 정지했으면 좀 풀어주지…-.-




Docker에서 컨테이너 조회 및 삭제 그리고 다시 이미지 삭제

아래와 같이 입력하면 현재 프로세스로 떠있는 모든 컨테이너의 목록을 확인한다.

docker ps -a



나는 현재 3개의 컨테이너가 프로세스로 등록되었고 모두 정지된 상태이며 사용하는 이미지는 모두 텐서플로우다.
가차없이 모두 죽여버리자!

docker rm $(docker ps -a -q)



다시 한 번 docker ps -a 명령을 실행하면 이제 목록에는 아무 것도 보이지 않는다.
이어서 바로 다시 한 번 이미지를 삭제해보자. 아래 그림과 같이 삭제 과정이 표시된다.





텐서플로우 이미지 설치

일단 1.0.0을 설치해야 하니까 뒤에 태그를 명시적으로 지정해주어야 할 것 같아서 아래와 같이 명령어를 입력했다.

docker run -it -p 8888:8888 -p 6006:6006 gcr.io/tensorflow/tensorflow:1.0.0-devel

음…그런데 이거 이미지 다 설치되고 나면 자동으로 jupyter가 실행되어야 하는데…이상하게 shell로 들어가버린다.
뭐냐? 이러면 나 흔들린다…ㅠ.ㅠ 여기서 뭘 어째야 하는거냐…?



일단 당황하지 않고~ 텐서플로우 홈페이지에 있었던 것처럼 다시 한 번 해본다.

docker run -it -p 8888:8888 -p 6006:6006 gcr.io/tensorflow/tensorflow

아하~! 별다른 태그를 주지 않으니까 자동으로 가장 최신의 버전이 설치되는군~
그리고 바로 Jupyter가 실행되면서 로그인에 필요한 토큰과 함께 Jupyter URL이 표시된다.



docker images를 통해 이미지 목록을 다시 한 번 조회해보면 2개의 이미지가 보인다. 위에 보이는 것은
나중에 설치한 이미지이고 아래 보이는 것은 1.0.0-devel 태그를 붙여 설치한 이미지이다.




최종 확인

마지막으로 Jupyter Notebook을 하나 만들어 다시 한 번 버전을 확인해보자.




이렇게 뭐로 가도 서울로 가면 된다고 어찌어찌 텐서플로우 1.0.0을 설치하였다~

반응형