Ubuntu14.04LTS에 Nvidia GPU driver (375.66) + CUDA 8.0 + CuDNN 6.0 설치하기

오랫동안 미루어왔던 머신러닝용 GPU설치를 마무리 하였다.
GPU설치 부분는 이제까지 많은 선배님들께서 가르침을 주시고 정리해 주셨다. 나는 그들의 가르침을 간단하게 셀스크립트로 정리하는 것으로 포스팅을 마무리 한다.

자세한 부분은 참고 링크를 통해서 선현님들께 직접 가르침을 받으시면 될 것 같다.

참고링크:

• http://flyingdcat4.tistory.com/76
• http://newsight.tistory.com/92
• http://kislayabhi.github.io/Installing_CUDA_with_Ubuntu/
• http://ejklike.github.io/2017/03/06/install-tensorflow1.0-on-ubuntu16.04-1.html
• Ndivia download link: https://developer.nvidia.com/accelerated-computing-toolkit
• My GITHUB: https://github.com/jwkanggist/DevEnvInstaller

My 설치 정보:
- Target OS : Ubuntu 14.04LTS
- GPU: Nvidia TitanX
- Nvidia GPU driver version: NVIDIA-Linux-x86_64-375.66
- CUDA version: cuda_8.0.61_375.26_linux
- CuDNN version: v6.0


준비 :
나의 셀 스크립트 실력이 부족하여 완전 원터치 설치를 구현하지 못했다. 그에 대한 장애물은 아래와 같다
- reboot 해야하는 부분
- conf파일을 edit해야 하는 부분
위에 대해서 쉘 스크립트 작성에 도움을 주실 수 있는 분은 조언 부탁드린다. 작성한 쉘스크립트 실행하기 전에 수동으로 실행해야할 명령어들은 아래와 같다.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15

# 1) Preliminary $ sudo apt-get install build-essential $ cd ~ $ mkdir ~/download/nvidia_installers # 2) GUI + X window off $ sudo rm /etc/X11/xorg.conf $ sudo service lightdm stop # 3) Deactivate nouveau driver (which is default driver for Ubuntu) $ sudo vim /etc/modprobe.d/blacklist-nouveau.conf # blacklist-nouveau.conf에서 아래를 수정 blacklist nouveau options nouveau modeset=0

엔비디아 드라이버 인스톨 (install_nvidiadriver-375.sh, root로 실행)

1 2 3 4 5 6 7 8 9 10

#! /bin/bash # Remove all nvidia related packages sudo apt-get --purge remove nvidia-* # Nvidia driver Installation # install nvidia driver add-apt-repository ppa:graphics-drivers/ppa sudo apt-get update sudo apt-get install nvidia-375 #recent version at 2017 June reboot

CUDA + CuDNN 인스톨 (install_cuda8.0.61_375.26_cudnn6.0.sh, root로 실행)

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28

#! /bin/bash # Nvidia driver installation check nvidia-smi # cuda 8.0 Installation wget https://developer.nvidia.com/compute/cuda/8.0/Prod2/local_installers/cuda_8.0.61_375.26_linux-run sh cuda_8.0.61_375.26_linux-run # Register env variables for CuDA echo -e "\n## CUDA and cuDNN paths" >> ~/.bashrc echo 'export PATH=/usr/local/cuda-8.0/bin:${PATH}' >> ~/.bashrc echo 'export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:${LD_LIBRARY_PATH}' >> ~/.bashrc # CUDA installation check source ~/.bashrc # CuDNN v.6.0 Installation # Nvidia site에서 수동으로 download sudo cp cuda/lib64/* /usr/local/cuda-8.0/lib64/ sudo cp cuda/include/* /usr/local/cuda-8.0/include/ sudo chmod a+r /usr/local/cuda-8.0/lib64/libcudnn* sudo chmod a+r /usr/local/cuda-8.0/include/cudnn.h # CuDNN installation check nvcc --version cat /proc/driver/nvidia/version cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2