티스토리 뷰
본 글은 Linux Ubuntu 16.04 LTS 64bit 기반으로 작성되었습니다.
TensorFlow를 설치하기 전에 우선 Docker를 리눅스에 설치해야합니다.
Docker에는 Community Edition(CE)버전과, Enterprise Edition(EE)버전이 있는데, 무료버전인 CE 버전을 이용하여 Docker를 먼저 설치하도록 하겠습니다.
터미널을 열고 아래의 명령어를 순서대로 입력하면 설치가 될 것입니다.
- Docker 설치를 위한 사전준비
1.apt package index를 업데이트 합니다.
1 | $sudo apt-get update | cs |
2.HTTPS를 이용하여 저장소를 받아올 수 있도록 추가적으로 패키지를 설치합니다.
1 2 3 4 5 | $sudo apt-get install \ > apt-transport-https \ > ca-certificates \ > curl \ software-properties-common | cs |
3. Docker의 공식 GPG key를 추가합니다.
1 | $curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add- | cs |
4.Docker amd64용 Stable 버전을 다운로드 받습니다.
1 2 3 4 | $ sudo add-apt-repository \ > "deb [arch=amd64] https://download.docker.com/linux/ubuntu \ > $(lsb_release -cs) \ > stable" | cs |
- Docker CE 버전 설치하기
1.apt package index를 업데이트 합니다.
1 | $sudo apt-get update | cs |
2.다운로드 받았던 Docker 프로그램을 설치합니다.
1 | $ sudo apt-get install docker-ce | cs |
3.설치가능한 Docker버전을 확인합니다
위 커맨드를 입력하면 설치 가능한 버전의 리스트가 여러 개 표시됩니다.
1 | docker-ce | 17.06.0~ce-0~ubuntu-xenial | https://download.docker.com/linux/ubuntu xenial/stable amd64 Packages | cs |
만약 위와 같은 버전의 docker를 설치하고 싶다면 아래의 커맨드를 입력합니다.
1 | sudo apt-get install docker-ce=17.06.0~ce-0~ubuntu-xenial | cs |
1 | $ sudo docker run hello-world | cs |
만약 위와 같은 결과가 나오지 않고 오류가 발생할 경우, 네트워크 환경을 점검해 보는것이 좋습니다.
- 텐서플로우 실행하기
아래 설명된 방법으로는 구글 클라우드 서버를 사용하여 작동되기 때문에 네트워크환경에만 사용할 수 있습니다.
텐서플로우는 CPU only버전과 GPU support 버전이 있습니다. 사용자의 환경에 따라서 적절한 버전을 선택하여 실행하시면 됩니다.
GPU Support버전은 NVIDIA의 CUDA Toolkit을 사용합니다. 아래 커맨드를 입력하여 CUDA 라이브러리를 설치하시기 바랍니다.
1 | $ sudo apt-get install libcupti-dev | cs |
CPU-only 버전은 아래 커맨드를 입력하여 실행합니다.
1 | $ docker run -it -p 8888:8888 gcr.io/tensorflow/tensorflow | cs |
GPU Support 버전은 아래 커맨드를 입력하여 실행합니다.
1 | $ nvidia-docker run -it -p 8888:8888 gcr.io/tensorflow/tensorflow:0.12.1-gpu | cs |
위 결과에 나온 URL을 FireFox 주소창에 복사 및 붙여넣기 해 주시면, 아래와 같은 페이지를 볼 수 있습니다. 이렇게 되면 성공적으로 텐서플로우를 실행한 것입니다.
- 텐서플로우에서 Hello World 실행해보기
텐서플로우에서 Python코드가 잘 실행되는지 테스트를 해 보겠습니다.
아래 커맨드를 입력하여 TensorFlow에서 실행시킬 Bash창을 엽니다.
1 | $ docker run -it gcr.io/tensorflow/tensorflow bash | cs |
Bash창을 연 뒤, 아래 코드를 입력합니다.
1 2 3 4 5 | # Python import tensorflow as tf hello = tf.constant('Hello, TensorFlow!') sess = tf.Session() print(sess.run(hello)) | cs |
위 코드의 결과로 Hello, TensorFlow! 가 나오면 정상적으로 실행된 것입니다.
앞으로 TensorFlow 작업을 하실때는
1 | $ docker run -it gcr.io/tensorflow/tensorflow bash | cs |
커맨드를 입력하신뒤 작업을 하시면 됩니다.
참고자료
https://docs.docker.com/engine/installation/linux/docker-ce/ubuntu/
https://www.tensorflow.org/versions/master/install/install_linux#top_of_page