쉬운 방법 : Ubuntu 18.04에 Nvidia 드라이버, CUDA, CUDNN 및 Tensorflow GPU 설치

tensorflow GPU를 쉽게 실행하려면 어떻게해야합니까?

우분투 18.04에 Tensorflow GPU를 설치했습니다. 그것에 대한 많은 지침이 있지만 가장 빠르고 쉬운 방법은 일반적으로 사용되지 않으며 공유하고 싶습니다.

엔비디아 드라이버 :

ubuntu-drivers devices
sudo ubuntu-drivers autoinstall
nvidia-smi

CUDA :

일반적으로 : "sudo apt install nvidia-cuda-toolkit"그러나 이것은 버전 9.1을 설치하지만 현재 너무 새로워 서 tensorflow가 실행되지 않습니다. 대신 다운로드 CUDA 9.0 : https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1704&target_type=runfilelocal 다음 실행 (추가 세부 사항은 첫 번째 줄에에서 찾을 수 있습니다 : 어떻게 할 수 우분투 17.10에 CUDA 9 설치 ) :

sudo sh cuda_9.0.176_384.81_linux.run --override
sudo ln -s /usr/bin/gcc-6 /usr/local/cuda/bin/gcc
sudo ln -s /usr/bin/g++-6 /usr/local/cuda/bin/g++
cd ~
sudo nano .bashrc

add at the end of the file:
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
CTL+X to save and exit

CuDNN :

nvidia 개발자 등록 https://developer.nvidia.com/cudnn 16.04 용 9.1 런타임 및 개발자 라이브러리 다운로드 (파일 cuDNN v7.1.3 Ubuntu16.04 용 런타임 라이브러리 (Deb) 및 cuDNN v7.1.3 Ubuntu16.04 용 개발자 라이브러리 ( Deb)) 소프트웨어 관리자로 파일을 열고 설치하십시오. 확인 :

cat /usr/include/x86_64-linux-gnu/cudnn_v*.h | grep CUDNN_MAJOR -A 2

라이브러리 및 tensorflow를 설치하십시오.

sudo apt-get install libcupti-dev
pip3 install tensorflow-gpu

검사:

in tensorflow check for GPU support 
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

참고 : 터미널에서 tensorflow 또는 개발 환경을 시작하십시오. 그렇지 않으면 PATH 변수가로드되지 않습니다.

Jonny의 대답을 확장하기 위해 : CUDA 설치를위한 Jonny의 방법을 따를 때 "Linux-x86_64 용 NVIDIA 가속 그래픽 드라이버 설치 384.81?"에 대해 "아니요"를 선택해야합니까? (다른 모든 것에 대해서는 예 또는 기본값). 그렇지 않으면 CUDA 드라이버 설치가 "오류 : NVIDIA 커널 모듈 'nvidia-drm'이 (가) 이미 커널에로드 된 것 같습니다"라는 메시지가 표시되지 않습니다.

그런 다음 샘플을 실행하여 CUDA 설치를 테스트 할 수 있습니다.

먼저 그들의 의존성을 설치하십시오 :

sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev libglfw3-dev libgles2-mesa-dev

이들은 Ubuntu 18의 새로운 컴파일러와 함께 컴파일되지 않으므로 gcc / g ++ 6으로 다운 그레이드해야합니다.

sudo apt-get install gcc-6 g++-6
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-6 10
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-6 10

다음을 탐색하여 샘플을 컴파일하십시오.

~/NVIDIA_CUDA-9.0_Samples/bin/x86_64/linux/release

그리고 다음을 사용하여 컴파일하십시오.

make -k

다음과 같은 샘플 중 하나를 실행하십시오.

./nbody