tensorflow (gpu support) 환경 셋팅하기

tensorflow (gpu support) 

vm (os : linux ubuntu 16)

nvidia gtx970m (laptap env) 

cuda 라이브러리 설치 (cuda toolkit 8.0 download link : https://developer.nvidia.com/cuda-downloads)

만약 나처럼 멍청이같이 9.1 latest버전을 설치했다면 cuda를 uninstall하고 8.0으로 다시깔자 

개무쓸모 도큐먼트 링크

https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#runfile-uninstallation

간단 지우는 커맨드 

sudo apt autoremove cuda

linux > x86_64 > Ubuntu > 16.04 > deb(network)

다운로드 후 아래 명령어 실행 (2018. 04월기준)

sudo dpkg -i cuda-repo-ubuntu1604_9.1.85-1_amd64.deb sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub sudo apt-get update sudo apt-get install cuda

cuDNN 및 환경변수 설정 

https://developer.nvidia.com/cudnn

nvidia 로그인이 필요하다 

버전은 cuDNN v6.0 (April 27, 2017), for CUDA 8.0

파일은 cuDNN v6.0 Library for Linux <<를 다운로드받는다 

tar xvzf cudnn-8.0-linux-x64-v6.0.tgz sudo cp -P cuda/include/cudnn.h /usr/local/cuda/include sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64 sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/include/cuda/lib64/libcudnn* sudo apt-get install libcupti-dev 환경변수 설정 (LD_LIBRARY_PATH, CUDA_HOME) vi ~/.bashrc export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} export CUDA_HOME=/usr/local/cuda

tensorflow-GPU 설치

pip3 install tensorflow-gpu