Простой способ: установить драйверы Nvidia, CUDA, CUDNN и Tensorflow GPU на Ubuntu 18.04

Ответы:

13

Я только что установил Tensorflow GPU на Ubuntu 18.04. Для этого есть много инструкций, однако я думаю, что самый быстрый и простой способ обычно не используется, и я хочу поделиться им:

NVIDIA DRIVER:

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

CUDA:

Обычно: "sudo apt install nvidia-cuda-toolkit". Однако это устанавливает версию 9.1, слишком новую на данный момент, и не будет запускаться тензор потока. Вместо того, чтобы загрузить 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 Затем запустите (Дополнительные детали на первой линии можно найти по адресу: Как Я устанавливаю CUDA 9 на 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:

Зарегистрируйтесь на сайте разработчиков nvidia https://developer.nvidia.com/cudnn Загрузите библиотеку времени выполнения 9.1 и библиотеку разработчика для 16.04 (файлы cuDNN v7.1.3 библиотека времени выполнения для Ubuntu16.04 (Deb) и библиотека cuDNN v7.1.3 для разработчика Ubuntu16.04 ( Деб)) Откройте файлы с помощью менеджера программного обеспечения и установите их. Проверить с:

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

Установите библиотеки и тензор потока:

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

Проверьте:

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

Примечание: Запустите tenorflow или вашу среду разработки из терминала, иначе для меня он не загружает переменные PATH.

Jonny
источник
есть ли способ заставить работать cuda 10.1?
j0h
на сайте nvidia я не увидел пакета cudnn для версии 9.1. 9.0,9.2,10.0,10.1 - это все, что я видел, о, и исходная версия
j0h
2

Просто чтобы расширить ответ Джонни: при следовании методу Джонни по установке CUDA мне пришлось выбрать «Нет» для «Установить драйвер ускоренной графики NVIDIA для Linux-x86_64 384.81?» (да или по умолчанию для всего остального). В противном случае установка драйвера CUDA завершится неудачно: «ОШИБКА: модуль ядра NVIDIA 'nvidia-drm', похоже, уже загружен в ваше ядро"

Затем вы можете проверить свою установку CUDA, запустив примеры.

Сначала установите их зависимости:

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

Они не будут компилироваться с новыми компиляторами в Ubuntu 18, поэтому вам нужно перейти на 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

Скомпилируйте образцы, перейдя к:

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

И скомпилировать, используя:

make -k

Запустите один из примеров, например:

./nbody
Роберт Корвус
источник