La manera fácil: instale controladores Nvidia, CUDA, CUDNN y GPU Tensorflow en Ubuntu 18.04

9

¿Cómo puedo hacer funcionar fácilmente tensorflow gpu?

18.04 cuda gpu

— Jonny
fuente

13

Acabo de instalar la GPU Tensorflow en Ubuntu 18.04. Hay muchas instrucciones para ello, sin embargo, creo que la forma más rápida y fácil generalmente no se usa y quiero compartirlo:

CONDUCTOR DE NVIDIA:

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

CUDA

Normalmente: "sudo apt install nvidia-cuda-toolkit" Sin embargo, esto instala la versión 9.1, demasiado nueva en este momento y no se ejecutará el tensorflow. En su lugar, descargue 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 Luego ejecute (se pueden encontrar detalles adicionales en la primera línea en: ¿Cómo se puede encontrar? Instalo CUDA 9 en 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:

regístrese en nvidia developers https://developer.nvidia.com/cudnn Descargue 9.1 biblioteca de tiempo de ejecución y desarrollador para 16.04 (Archivos cuDNN v7.1.3 Biblioteca de tiempo de ejecución para Ubuntu16.04 (Deb) y cuDNN v7.1.3 Biblioteca de desarrollador para Ubuntu16.04 ( Deb)) Abra los archivos con el administrador de software e instálelos. Comprobar con:

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

Instalar bibliotecas y tensorflow:

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

Cheque:

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

Nota: Inicie tensorflow o su entorno de desarrollo desde la terminal, de lo contrario para mí no carga las variables PATH.

— Jonny
fuente

¿hay alguna forma de hacer que cuda 10.1 funcione?

— j0h

en el sitio web de nvidia, no vi ningún paquete cudnn para la versión 9.1. 9.0,9.2,10.0,10.1 es todo lo que vi, oh, y una versión fuente

— j0h

2

Solo para extender la respuesta de Jonny: al seguir el método de Jonny para instalar CUDA, tuve que elegir "No" para "¿Instalar NVIDIA Accelerated Graphics Driver para Linux-x86_64 384.81?" (sí o por defecto para todo lo demás). De lo contrario, la instalación del controlador CUDA fallaría diciendo "ERROR: Parece que un módulo de kernel de NVIDIA 'nvidia-drm' ya está cargado en su kernel"

Luego puede probar su instalación de CUDA ejecutando las muestras.

Primero instale sus dependencias:

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

Estos no se compilarán con los nuevos compiladores en Ubuntu 18, por lo que deberá cambiar 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

Compile las muestras navegando a:

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

Y compilar usando:

make -k

Ejecute una de las muestras como:

./nbody

— Robert Corvus
fuente