TensorFlow

공식 홈페이지의 Download and Setup을 참고하자.

요구사항

  1. NVIDIA CUDA Toolki >= 7.0
  2. NVIDIA cuDNN Library >= v2
  3. Python 2.7 (or 3.3+)
  4. (Optional)Bazel

TensorFlow 설치

먼저 pip와 python-dev가 설치되어 있지 않다면 설치한다. pip를 이용하여 아래와 같이 설치할 수 있다. 우리는 GPU를 사용할 것이므로, 두 번째 방법으로 설치한다.

# Ubuntu/Linux 64-bit, CPU only:
$ sudo pip install --upgrade https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.7.0-py2-none-linux_x86_64.whl

# Ubuntu/Linux 64-bit, GPU enabled:
$ sudo pip install --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.7.0-py2-none-linux_x86_64.whl