Ubuntu Tensorflow 설치

CUDA 설치 방법: https://deeptak.tistory.com/2

Anaconda 설치 방법: https://deeptak.tistory.com/3

지난 포스팅을 이용하여 Ubuntu 환경에서 CUDA & Anaconda를 설치하였다.

 

1) CUDA 버전 확인하기

Tensorflow를 설치하기 이전 현재 CUDA 버전이 몇인지 먼저 확인해야 한다.

$ nvcc --version

 

 

아래 표를 통해 Tensorflow를 설치 이전 현재 CUDA에 호환이 되는 버전을 확인해야한다.

 

호환 되는 버전을 확인 하였으면 Anaconda 가상 환경을 실행하고, 텐서플로우를 설치해본다.

 

2) Tensorflow 설치하기

$ source activate EnvaName
$ pip install tensorflow-gpu==1.14.0

주의) 뒤에 -gpu를 붙이지 않고 pip install tensorflow를 실행하는 경우 CPU 기반의 tensorflow 라이브러리가 설치된다. 

 

3) 설치 확인방법

python을 실행시키고 해당 명령어를 수행해본다. 

import tensorflow as tf
tf.__version__

Tensorflow 버전이 출력되면 정상적으로 설치된 것이다.