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

Respuestas:

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