TensorFlow 설치 (우분투 14.04 64bit, 2016-05-16)


2016년 5월 16일 기준으로 TensorFlow를 아래와 같이 설치를 진행하였다.
우분투 14.04 64bit에 설치를 진행하였다
기록 차원에서 아래와 같이 정리한다

1. GPU 기능을 활용하기 위해서 nvidia 드라이버를 설치한다

  • $ sudo apt-get install nvidia-current nvidia-settings

2. Python 개발 환경을 Anaconda로 설치한다
Anaconda 사이트에서 설치 파일을 다운로드 후 설치를 진행한다
(https://www.continuum.io/downloads#linux)
Ubuntu 14.04 64비트에는 python이 디폴트로 2.7 버전이 설치되어 있으므로
Python 2.7버전을 다운 받아서 설치한다
  • $ bash ./Anaconda2-4.0.0-Linux-x86_64.sh
3. cuda 패키지를 설치한다
  • $ sudo dpkg -i ./cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb 
  • $ sudo apt-get update
  • $ sudo apt-get install cuda
  • $ sudo apt-get install nvidia-cuda-toolkit
4. bashrc에 아래 설정을 추가한다
  • export CUDA_HOME=/usr/local/cuda-7.5 
  • export LD_LIBRARY_PATH=${CUDA_HOME}/lib64 
  • export PATH=${CUDA_HOME}/bin:${PATH} 
5. cudnn을 다운로드 후 include와 lib을 cuda 설치 위치로 복사한다
  • cudnn-7.0-linux-x64-v4.0-prod.tgz
  • copy /usr/local/lib64 include
6. TensorFlow 설치를 한다
  • $ source ~/anaconda2/bin/activate ~/anaconda2/
  • (/home/dkyun77/anaconda2)[~]#
  • # Ubuntu/Linux 64-bit, GPU enabled, Python 2.7
  • # Requires CUDA toolkit 7.5 and CuDNN v5. For other versions, see "Install from sources" below.
  • (/home/dkyun77/anaconda2)[~]# export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0rc0-cp27-none-linux_x86_64.whl
  • (/home/dkyun77/anaconda2)[~]# sudo pip install --upgrade $TF_BINARY_URL
7. TensorFlow 설치를 확인한다
  • $ source anaconda2/bin/activate ~/anaconda2/
  • (/home/dkyun77/anaconda2)[~]# ipython notebook
    • ...
  • jupyter 환경에서 "import tensorflow"를 실행해서 에러가 없으면 정상 설치된 것이다





댓글

이 블로그의 인기 게시물

SSH 연결 Delay 해결

[ELK] search guard를 이용한 보안 설정 (사용자 권한)

공공데이터(openapi) 사용법 (특정 정류소, 버스의 남은 좌석 확인 하기)