Ubuntu 18.04, CUDA 10.1, libcudnn 7.5.1 и драйверы NVIDIA 418.67

Примечания

• 2019-06-23 : Последние обновления с версиями CUDA 10.0 или 10.1 драйвер NVIDIA 418.67, который устанавливается вместе с ним, больше не включает 32-битные библиотеки, и это будет приведет к тому, что Steam и большинство игр больше не будут работать. Версия libnvidia-gl-418: i386 устанавливает только версию 418.56, которая не будет работать с драйвером 418.67. Надеюсь, NVIDIA скоро выпустит обновление для этого. Я добавил информацию внизу этого ответа в .run file install часть того, как загрузить только файл запуска для установщика CUDA, после чего вы можете использовать любой драйвер, который хотите. Размер исполняемого файла составляет 2,3 ГБ, поэтому его загрузка может занять немного времени.

• CUDA 9.x недоступен в репозитории NVIDIA ubuntu1804. Однако я написал ответ для CUDA 9.2 по адресу https://askubuntu.com/a/1086993/231142

Установка CUDA через репозиторий (вместо установки .deb )

Следующие строки вы можете скопировать и вставить в окно терминала. Нажмите Ctrl + Alt + T , чтобы открыть окно терминала.

Удалить и обновить

Удалите все CUDA PPA, которые могут быть настроены, а также удалить nvidia-cuda-toolkit , если установлен:

sudo rm /etc/apt/sources.list.d/cuda*
sudo apt remove --autoremove nvidia-cuda-toolkit

Рекомендуется также удалить все драйверы NVIDIA перед установкой новых драйверов:

sudo apt remove --autoremove nvidia-*

Затем обновить систему:

sudo apt update

Добавить и установить

Недавно я только что выяснил, что установка CUDA работает с графическими драйверами ppa , поэтому, если вы не добавили его, добавьте его сейчас:

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update

Установите драйвер NVIDIA. Для этого мы собираемся использовать драйвер 440

sudo apt install nvidia-driver-440

Теперь установите ключ:

sudo apt-key adv --fetch-keys  http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub

Добавьте репозитории:

sudo bash -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/cuda.list'

sudo bash -c 'echo "deb http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/cuda_learn.list'

Обновите систему снова:

sudo apt update

Установите CUDA 10.1:

sudo apt install cuda-10-1

Должна быть установлена ​​NVIDIA 418.40 с указанными в репо драйверами. См .: http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/

Установите libcudnn7 7.5.1:

sudo apt install libcudnn7

Настройте среду и перезагрузите

Добавьте следующие строки в ваш файл ~ / .profile для CUDA 10.1

# set PATH for cuda 10.1 installation
if [ -d "/usr/local/cuda-10.1/bin/" ]; then
    export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}
    export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
fi

Перезагрузите компьютер.

Проверьте настройки

Проверьте компилятор NVIDIA Cuda с помощью nvcc --version :

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Apr_24_19:10:27_PDT_2019
Cuda compilation tools, release 10.1, V10.1.168

Проверить версию libcudnn / sbin / ldconfig -N -v $ (sed 's /: / /' <<< $ LD_LIBRARY_PATH) 2> / dev / null | grep libcudnn :

terrance@terrance-ubuntu:~$ /sbin/ldconfig -N -v $(sed 's/:/ /' <<< $LD_LIBRARY_PATH) 2>/dev/null | grep libcudnn
    libcudnn.so.7 -> libcudnn.so.7.5.1

Проверьте драйвер NVIDIA с помощью nvidia-smi :

terrance@terrance-ubuntu:~$ nvidia-smi
Wed Jan 29 12:41:02 2020       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 440.48.02    Driver Version: 440.48.02    CUDA Version: 10.2     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 750 Ti  Off  | 00000000:02:00.0  On |                  N/A |
| 40%   34C    P0     1W /  38W |    163MiB /  2000MiB |      1%      Default |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0      1993      G   /usr/lib/xorg/Xorg                           158MiB |
|    0      2502      G   compton                                        1MiB |
+-----------------------------------------------------------------------------+

.run файл install

Установите драйвер

Используя sudo add- apt-repository ppa: graphics-drivers / ppa вы можете установить новейший драйвер 430.26 или любой другой, который вам подходит.

Установите libcudnn7

Добавьте репо:

sudo bash -c 'echo "deb http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/cuda_learn.list'

Установите ключ:

sudo apt-key adv --fetch-keys  http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub

Обновите систему:

sudo apt update

Установите libcudnn7.5.1:

sudo apt install libcudnn7

Загрузите файл .run

Теперь загрузите cuda_10.1.105_418.39_linux.run из https://developer.nvidia.com/cuda-10.1-download-archive-base?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=runfilelocal

Если вам нужны инструкции по загрузке CUDA, здесь 10.2. https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=runfilelocal Затем выполните те же действия, что и в b elow, но обязательно обновите до 10.2 вместо 10.1.

Затем запустите установщик:

sudo sh cuda_10.1.105_418.39_linux.run

Введите accept и нажмите Enter на этом экране:

┌──────────────────────────────────────────────────────────────────────────────┐
│  End User License Agreement                                                  │
│  --------------------------                                                  │
│                                                                              │
│                                                                              │
│  Preface                                                                     │
│  -------                                                                     │
│                                                                              │
│  The Software License Agreement in Chapter 1 and the Supplement              │
│  in Chapter 2 contain license terms and conditions that govern               │
│  the use of NVIDIA software. By accepting this agreement, you                │
│  agree to comply with all the terms and conditions applicable                │
│  to the product(s) included herein.                                          │
│                                                                              │
│                                                                              │
│  NVIDIA Driver                                                               │
│                                                                              │
│                                                                              │
│  Description                                                                 │
│                                                                              │
│  This package contains the operating system driver and                       │
│──────────────────────────────────────────────────────────────────────────────│
│ Do you accept the above EULA? (accept/decline/quit):                         │
│ accept                                                                       

Снимите выделение с драйвера и затем выберите «Установить» с помощью клавиш со стрелками и пробела панель для перемещения и выбора или отмены выбора:

┌──────────────────────────────────────────────────────────────────────────────┐
│ CUDA Installer                                                               │
│ - [ ] Driver                                                                 │
│      [ ] 418.39                                                              │
│ + [X] CUDA Toolkit 10.1                                                      │
│   [X] CUDA Samples 10.1                                                      │
│   [X] CUDA Demo Suite 10.1                                                   │
│   [X] CUDA Documentation 10.1                                                │
│   Install                                                                    │
│   Options                                                                    │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│ Up/Down: Move | Left/Right: Expand | 'Enter': Select | 'A': Advanced options │

Дождитесь завершения установки, во время установки могут появиться сообщения об ошибках, но не беспокойтесь.

Настройте среду и перезагрузите

Добавьте следующие строки в свой ~ / Файл .profile для CUDA 10.1

# set PATH for cuda 10.1 installation
if [ -d "/usr/local/cuda-10.1/bin/" ]; then
    export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}
    export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
fi

Перезагрузите систему, чтобы изменения вступили в силу.

Драйверы Ubuntu 16.04, CUDA 9.2 и NVIDIA 396:

ПРИМЕЧАНИЕ: Репозиторий NVIDIA решил продвинуть драйверы 410. Я проведу небольшое тестирование, чтобы увидеть, смогу ли я настроить его для драйвера, который вы хотите установить.

Эти инструкции предназначены для установки CUDA через репозиторий вместо установки .deb .

следующие строки вы можете скопировать и вставить в окно терминала. Нажмите Ctrl + Alt + T , чтобы открыть окно терминала.

Удалите все CUDA PPA, которые могут быть настроены, а также удалите nvidia- cuda-toolkit , если установлен:

sudo rm /etc/apt/sources.list.d/cuda*
sudo apt remove nvidia-cuda-toolkit

Рекомендуется также удалить все драйверы NVIDIA перед установкой новых драйверов:

sudo apt remove nvidia-*

Затем обновить систему:

sudo apt update

Установить ключ:

sudo apt-key adv --fetch-keys  http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub

Добавить репо:

sudo bash -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64 /" > /etc/apt/sources.list.d/cuda.list'

Обновить для добавляемого нового репо:

sudo apt update

Установить CUDA 9.2:

sudo apt install cuda-9-2

С ним должны быть установлены драйверы nvidia-396, так как они перечислены в репо. См .: http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/

Убедитесь, что CUDA 9.2 был установлен:

~$ ls /usr/local/cuda-9.2/
bin     include    libnvvp           nvml    samples  targets
doc     lib64      LICENSE           nvvm    share    tools
extras  libnsight  nsightee_plugins  README  src      version.txt

Теперь добавьте следующее в свой ~ / .profile для PATH и LD_LIBRARY . Вы можете использовать команду gedit ~ / .profile для редактирования:

# set PATH for cuda 9.2 installation
if [ -d "/usr/local/cuda-9.2/bin/" ]; then
    export PATH=/usr/local/cuda-9.2/bin${PATH:+:${PATH}}
    export LD_LIBRARY_PATH=/usr/local/cuda-9.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
fi

Перезагрузите систему.

sudo reboot

Когда система будет запущена, вы можете проверить установку, введя следующее:

nvcc -V

Вы должно появиться что-то похожее на следующее:

~$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Nov__3_21:07:56_CDT_2017
Cuda compilation tools, release 9.1, V9.1.85

И вы должны увидеть установленные драйверы 396.xx :

~$ nvidia-smi
Thu May 17 07:38:54 2018        
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 396.44                 Driver Version: 396.44                    |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 760     Off  | 00000000:02:00.0 N/A |                  N/A |
| 49%   53C    P0    N/A /  N/A |    187MiB /  1999MiB |     N/A      Default |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0                    Not Supported                                       |
+-----------------------------------------------------------------------------+

Ubuntu 14.04, CUDA 10.0 и NVIDIA 410 драйверов:

Обратите внимание на то, что мне не установили Ubuntu 14.04, таким образом, я не могу проверить, работают ли эти шаги или нет. Но согласно http://developer.download.nvidia.com/compute/cuda/repos/ от NVIDIA, это должны быть те же шаги, как я перечислил в других двух ответах.

Эти инструкции для установки CUDA через репозиторий вместо .deb установка.

Следующие строки можно скопировать и вставить к окну терминала. Нажмите Ctrl+Alt+T для открытия окна терминала.

Удалите любой CUDA PPAs, который может быть установкой и также удалить nvidia-cuda-toolkit если установлено:

sudo rm /etc/apt/sources.list.d/cuda*
sudo apt remove nvidia-cuda-toolkit

Рекомендуемый также удалить все драйверы NVIDIA прежде, чем установить новые драйверы:

sudo apt remove nvidia-*

Затем обновите систему:

sudo apt update

Установите ключ:

sudo apt-key adv --fetch-keys  http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/7fa2af80.pub

Добавьте repo:

sudo bash -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64 /" > /etc/apt/sources.list.d/cuda.list'

Обновите систему снова:

sudo apt update

CUDA 10.0 установки.

sudo apt install cuda-10-0

Это должно устанавливать Nvidia 410 драйверов с ним, поскольку это - то, что перечислено в repo. См.: http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/

Добавьте следующие строки к Вашему ~/.profile файл для CUDA 10.0

# set PATH for cuda 10.0 installation
if [ -d "/usr/local/cuda-10.0/bin/" ]; then
    export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
    export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
fi

Перезагрузите компьютер и проверьте свои настройки, когда перезагрузка будет завершена:

Проверьте компилятор NVIDIA Cuda с nvcc --version:

Проверьте драйвер NVIDIA с nvidia-smi:

Не ошибитесь .

Загрузите файл .deb раз и навсегда.

Я скачал файл .run и установил его. Но как только я попытался установить тензоррт, оказалось, что его нельзя установить вместе с ним.

Я смог установить его только после того, как был установлен драйвер .deb nvidia.

Для всех, кто работает с глубокой обучающей базой AWS AMI:
. Образ поставляется с несколькими встроенными окружениями NVIDIA CUDA и по умолчанию работает с одним из них (в моем случае 10), но вы можете переключиться на любое другое, просто изменив сим-ссылку:

$ sudo rm /usr/local/cuda
$ sudo ln -s /usr/local/cuda-10.2 /usr/local/cuda

Подробнее:
https://docs.aws.amazon.com/dlami/latest/devguide/tutorial-base.html