Prosty sposób: zainstaluj sterowniki Nvidia, CUDA, CUDNN i procesor graficzny Tensorflow na Ubuntu 18.04

Jak mogę łatwo uruchomić tensorflow GPU?

Właśnie zainstalowałem procesor graficzny Tensorflow na Ubuntu 18.04. Jest na to wiele instrukcji, ale myślę, że najszybszy i najłatwiejszy sposób zwykle nie jest używany i chcę się nim podzielić:

STEROWNIK NVIDIA:

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

CUDA:

Zwykle: „sudo apt install nvidia-cuda-toolkit” Jednak to instaluje wersję 9.1, w tej chwili zbyt nowa i tensorflow nie będzie działał. Zamiast pobrać 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 Następnie uruchom (Dodatkowe szczegóły w pierwszym wierszu można znaleźć pod adresem: Jak można Instaluję CUDA 9 na Ubuntu 17.10 ):

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:

zarejestruj się w nvidia developers https://developer.nvidia.com/cudnn Pobierz środowisko uruchomieniowe i bibliotekę programistów 9.1 dla 16.04 (Pliki cuDNN v7.1.3 Runtime Library dla Ubuntu16.04 (Deb) i cuDNN v7.1.3 Biblioteka programistów dla Ubuntu16.04 ( Deb)) Otwórz pliki za pomocą menedżera oprogramowania i zainstaluj je. Sprawdź z:

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

Zainstaluj biblioteki i tensorflow:

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

Czek:

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

Uwaga: Uruchom tensorflow lub środowisko programistyczne z terminala, w przeciwnym razie dla mnie nie ładuje zmiennych PATH.

Aby rozszerzyć odpowiedź Jonny'ego: postępując zgodnie z metodą Jonny'ego dotyczącą instalacji CUDA, musiałem wybrać „Nie” dla opcji „Zainstaluj sterownik karty graficznej NVIDIA dla systemu Linux-x86_64 384,81?” (tak lub domyślnie dla wszystkich innych). W przeciwnym razie instalacja sterownika CUDA nie powiedzie się „BŁĄD: moduł jądra NVIDIA„ nvidia-drm ”wydaje się być już załadowany do twojego jądra”

Następnie możesz przetestować instalację CUDA, uruchamiając próbki.

Najpierw zainstaluj ich zależności:

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

Nie będą się kompilować z nowymi kompilatorami na Ubuntu 18, więc musisz obniżyć wersję do 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

Skompiluj próbki, przechodząc do:

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

I kompiluj używając:

make -k

Uruchom jedną z próbek, takich jak:

./nbody