Nvidia Driver 제거

$ sudo /usr/bin/nvidia-uninstall # 사용하던 nvidia 드라이버가 있으면 제거

CUDA 설치

http://docs.nvidia.com/cuda/cuda-installation-guide-linux/#axzz4VZnqTJ2A 에서 다운로드 하여 설치

CUDA는 sample까지 설치후 sample 데릭토리에서 build 하여 샘플프로그램이 정상 실행되는지 확인하다.

CUDNN 설치

https://developer.nvidia.com/cudnn에서 다운받아 설치

before build

$ sudo apt-get install build-essential

환경변수 추가

본인이 설치한 환경에 맞추어 환경변수를 설정하자.

$ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/lks21c/cudnn_60/lib64
$ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64

virtual env 준비

$ sudo apt-get install python-pip python-dev python-virtualenv

virtualenv 생성

$ virtualenv --system-site-packages tf27

virtualenv activate

$ source ~/tf27/bin/activate

python 2.7 + gpu

$ pip install --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.1.0-cp27-none-linux_x86_64.whl

직접 소스 빌드하여 설치

직접 빌드하여 설치하는 포스팅을 참고하자.