ubuntu 16.04에 tensorflow 설치하기(GPU)

안녕하세요? tensorflow gpu 버전 설치 방법에 대해 설명해보도록 하겠습니다.

시스템 정보:

프로세서:Intel core i5-7600 3.50GHz x 4

그래픽:GeForce GTX 1070/PCle/SSE2

OS 종류: ubuntu 64bit

일단, tensorflow gpu 버전을 사용하기 위해서는 cuda와 cudnn의 설치가 필요합니다.

위 두개에 관한 설치는 다른 페이지에 이미 소개를 해놨기 때문에, 참고하시길 바랍니다.

저 같은 경우에는 cuda 8과 cudnn 5.1을 사용했습니다.

경로는 vi ~/.bashrc에 들어가서 입력시켰습니다.

참고하시길 바랍니다.

본격적으로 tensorflow gpu 설치에 대해 설명하겠습니다.

저는 virtualenv로 설치했습니다. 

-pip 과 virtualenv를 설치합니다.

-디렉토리 ~/tensorflow에 virtualenv 환경을 만듭니다.

-환경을 활성화합니다.

-이제 pip 설치 방식과 동일학 텐서플로우를 설치합니다. 먼저 적절한 바이너리를 선택합니다:

# Ubuntu/Linux 64-bit, CPU 전용, Python 2.7
(tensorflow)$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.9.0-cp27-none-linux_x86_64.whl

# Ubuntu/Linux 64-bit, GPU 버전, Python 2.7
# CUDA toolkit 7.5 와 CuDNN v4 필수. 다른 버전을 사용하려면 아래 "소스에서 설치" 섹션을 참고하세요.
(tensorflow)$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.9.0-cp27-none-linux_x86_64.whl

# Mac OS X, CPU 전용, Python 2.7
(tensorflow)$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/mac/tensorflow-0.9.0-py2-none-any.whl

# Ubuntu/Linux 64-bit, CPU 전용, Python 3.4
(tensorflow)$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.9.0-cp34-cp34m-linux_x86_64.whl

# Ubuntu/Linux 64-bit, GPU 버전, Python 3.4
# CUDA toolkit 7.5 와 CuDNN v4 필수. 다른 버전을 사용하려면 아래 "소스에서 설치" 섹션을 참고하세요.
(tensorflow)$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.9.0-cp34-cp34m-linux_x86_64.whl

# Ubuntu/Linux 64-bit, CPU 전용, Python 3.5
(tensorflow)$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.9.0-cp35-cp35m-linux_x86_64.whl

# Ubuntu/Linux 64-bit, GPU 버전, Python 3.5
# CUDA toolkit 7.5 와 CuDNN v4 필수. 다른 버전을 사용하려면 아래 "소스에서 설치" 섹션을 참고하세요.
(tensorflow)$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.9.0-cp35-cp35m-linux_x86_64.whl

# Mac OS X, CPU 전용, Python 3.4 or 3.5:
(tensorflow)$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/mac/tensorflow-0.9.0-py3-none-any.whl

저는 python 3.5버전을 사용했습니다.

마지막으로 텐서플로우를 설치합니다:

$TF_BINARY_URL은 위의 그림에서 자기 버전에 맞는 것을 골라서 바꿔서 넣으면됩니다. 예를 들어, 저같은 경우는 아래그림과 같습니다.

마지막으로, 아래와 같이 테스트해보시면 비슷한 결과가 나올겁니다.

끝으로, 텐서플로우 작업을 마쳤을 때는 환경을 비활성화 합니다.

나중에 텐서플로우를 다시 사용하려면 다시 활성화 하면 됩니다.