¿Cómo puedo hacer funcionar fácilmente tensorflow gpu?
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.
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:
Estos no se compilarán con los nuevos compiladores en Ubuntu 18, por lo que deberá cambiar a gcc / g ++ 6:
Compile las muestras navegando a:
Y compilar usando:
Ejecute una de las muestras como:
fuente