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

9

Як мені легко запустити tensorflow gpu?

18.04 cuda gpu

— Джоні
джерело

13

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

Драйвер NVIDIA:

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

CUDA:

Зазвичай: "sudo apt install nvidia-cuda-toolkit" Однак при цьому встановлюється версія 9.1, занадто нова на даний момент, і tensorflow не запускається. Замість цього завантажте 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

Встановіть бібліотеки та tensorflow:

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))

Примітка. Почніть tensorflow або ваше середовище розробки з терміналу, інакше для мене він не завантажує змінні PATH.

— Джоні
джерело

чи все-таки потрібно заробляти cuda 10.1?

— j0h

На веб-сайті nvidia я не побачив жодного пакета cudnn для версії 9.1. 9.0,9.2,10.0,10.1 - це все, що я бачив, о, та версія джерела

— j0h

2

Просто для розширення відповіді Джоні: коли дотримувався методу Джонні для встановлення CUDA, мені довелося вибрати "Ні" для "Встановити NVIDIA Accelerated Graphics Driver для 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

— Роберт Корвус
джерело