Простой способ: установить драйверы Nvidia, CUDA, CUDNN и Tensorflow GPU на Ubuntu 18.04
2 ответа
Я просто установил GPU Tensorflow на Ubuntu 18.04. Существует много инструкций для него, однако я думаю, что самый быстрый и самый легкий путь обычно не используется, и я хочу совместно использовать его:
ДРАЙВЕР NVIDIA:
ubuntu-drivers devices
sudo ubuntu-drivers autoinstall
nvidia-smi
CUDA:
Обычно: "sudo способная установка nvidia-cuda-toolkit" Однако это устанавливает версию 9.1, слишком новую в данный момент, и tensorflow не будет работать. Вместо этого загрузите CUDA 9.0: https://developer.nvidia.com/cuda-90-download-archive? al Then target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1704&target_type=runfileloc работает (Дополнительные Детали о первой строке могут быть найдены в: Как может я устанавливать 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 (Deb)) Открывает файлы с менеджером по программному обеспечению и устанавливает их. Сверьтесь:
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 или Вашу среду разработки от терминала, иначе для меня это не загружает переменные ПУТИ.
Только расширить ответ Jonny: при следовании методу Jonny для установки CUDA я должен был выбрать "No" для "Установки NVIDIA Ускоренный Графический драйвер для Linux-x86_64 384.81?" (да или значение по умолчанию для всего остального). Иначе установка драйвера CUDA привела бы высказывание к сбою "ОШИБКА: модуль ядра NVIDIA 'DRM Nvidia', кажется, уже загружается в Вашем ядре"
Затем можно протестировать установку 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