Il modo più semplice: installa i driver Nvidia, CUDA, CUDNN e GPU Tensorflow su Ubuntu 18.04

9

Come posso far funzionare facilmente tensorflow gpu?

18.04 cuda gpu

— Jonny
fonte

13

Ho appena installato la GPU Tensorflow su Ubuntu 18.04. Ci sono molte istruzioni per questo, tuttavia penso che il modo più semplice e veloce di solito non sia usato e voglio condividerlo:

DRIVER NVIDIA:

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

CUDA:

Normalmente: "sudo apt install nvidia-cuda-toolkit" Tuttavia, questa installa la versione 9.1, al momento troppo nuova e tensorflow non funzionerà. Invece scaricare 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 Quindi eseguire (Dettagli aggiuntivi sulla prima linea è disponibile all'indirizzo: Come può Installo CUDA 9 su 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:

registrati presso gli sviluppatori nvidia https://developer.nvidia.com/cudnn Scarica la libreria runtime 9.1 e gli sviluppatori per 16.04 (Files cuDNN v7.1.3 Libreria runtime per Ubuntu16.04 (Deb) e cuDNN v7.1.3 Developer Library per Ubuntu16.04 ( Deb)) Apri i file con il software manager e installali. Controllare con:

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

Installa librerie e tensorflow:

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

Dai un'occhiata:

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

Nota: avviare tensorflow o l'ambiente di sviluppo dal terminale, altrimenti per me non carica le variabili PATH.

— Jonny
fonte

c'è qualche modo per far funzionare cuda 10.1?

— j0h

sul sito di nvidia, non ho visto alcun pacchetto cudnn per la versione 9.1. 9.0.9.2,10.0,10.1 è tutto ciò che ho visto, oh, e una versione sorgente

— j0h

2

Solo per estendere la risposta di Jonny: quando ho seguito il metodo di Jonny per l'installazione di CUDA, ho dovuto scegliere "No" per "Installa il driver NVIDIA Accelerated Graphics per Linux-x86_64 384.81?" (sì o predefinito per tutto il resto). Altrimenti l'installazione del driver CUDA fallirebbe dicendo "ERRORE: un modulo del kernel NVIDIA 'nvidia-drm' sembra essere già caricato nel tuo kernel"

Quindi è possibile verificare l'installazione di CUDA eseguendo gli esempi.

Prima installa le loro dipendenze:

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

Questi non verranno compilati con i nuovi compilatori su Ubuntu 18, quindi dovrai effettuare il downgrade a 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

Compilare gli esempi andando a:

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

E compilare usando:

make -k

Esegui uno dei campioni come:

./nbody

— Robert Corvus
fonte