Ubuntu 18.04 NVIDIA 드라이버 설치, CUDA 설정, Tensorflow 프로 세 스

참조 링크
http://homeway.me/2017/08/08/setup-gpu-for-tensorflow/
 
설치 중 루트 로 로그 인
글 의 일부 패 키 지 는 자신의 상황 에 따라 다운로드 해 야 하 며 바보 식 조작 이 아니다.필요 하면 메 시 지 를 남기 거나 사적인 편 지 를 남 길 수 있 습 니 다.보 내 드릴 수 있 습 니 다. 사랑 하 는 손님 에 게 밥값 을 줄 수 있 습 니 다 ~ ~.
 
1. 그래 픽 카드 설정 보기

lspci | grep -i vga

 
2. 해당 하 는 그래 픽 카드 드라이버 다운로드
https://www.nvidia.com/Download/index.aspx?lang=en-us
Nvidia 구동 을 위 한 의존 환경 구축 

sudo apt-get install dkms build-essential linux-headers-generic


sudo mkdir /etc/modprobe.d/

sudo touch /etc/modprobe.d/blacklist-nouveau.conf


cat >> /etc/modprobe.d/blacklist-nouveau.conf << EOF

blacklist nouveau

blacklist lbm-nouveau

options nouveau modeset=0

alias nouveau off

alias lbm-nouveau off

EOF

sudo vi blacklist - nouveau. conf 를 읽 을 수 없다 면

echo options nouveau modeset=0 | sudo tee -a /etc/modprobe.d/nouveau-kms.conf

sudo update-initramfs -u


sudo reboot

NVIDIA - linux - XXXX 는 대응 하 는 그래 픽 카드 구동 버 전 으로 수정 이 필요 합 니 다.

sudo chmod u+x NVIDIA-Linux-x86_64-390.77.run

sudo ./NVIDIA-Linux-x86_64-390.77.run

sudo reboot

3. cuda 설치

sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb

sudo apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub

sudo apt-get update

sudo apt-get install cuda -y


CUDA_HOME=/usr/local/cuda-10.0

ln -vfs $CUDA_HOME /usr/local/cuda

cat >> ~/.profile << EOF

export CUDA_HOME=${CUDA_HOME}

export LD_LIBRARY_PATH=${CUDA_HOME}/lib64

PATH=${CUDA_HOME}/bin:${PATH}

EOF

source ~/.profile

이 경험 치 는 우리 에 게 다시 시작 해 야 한 다 는 것 을 알려 주 었 다. reboot.
 
테스트 CUDA
➜ $ cd $CUDA_HOME/samples/1_Utilities/deviceQuery
➜ $ make
➜ $ ./deviceQuery
 
4. CUDA NN 설치

tar xzvf cudnn-8.0-linux-x64-v5.1.tgz

sudo cp cuda/include/cudnn.h /usr/local/cuda/include

sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

5. Python 설치 및 의존 환경

sudo apt-get install python-dev python3-dev python-numpy \

 python3-numpy python-six python3-six build-essential python-pip python3-pip


sudo dpkg -i cuda-repo-ubuntu1604_9.1.85-1_amd64.deb

sudo dpkg -i nvidia-machine-learning-repo-ubuntu1604_1.0.0-1_amd64.deb

sudo apt-get update


dpkg -i  nccl-repo-ubuntu1804-2.3.5-ga-cuda10.0_1-1_amd64.deb

apt-get update

6. 텐서 플 로 우 설치
pip3 install tensorflow-1.12.0rc0-cp36-cp36m-linux_x86_64.whl
numpy 를 다시 달 아야 돼 요.
pip3 install -U numpy