Установка CUDA

Правильная установка CUDA и драйвера NVIDIA на вашем компьютере может быть сложной задачей. Это руководство должно рассказать вам всё, что вам нужно знать (по крайней мере, для Ubuntu!).

Ещё один хороший ресурс — это руководство по установке CUDA для Linux.

GPU

Вам понадобится графический процессор NVIDIA, если вы хотите использовать CUDA. Если вы хотите использовать cuDNN, вам потребуется GPU с вычислительной мощностью >= 3.0. Чтобы узнать, какова вычислительная мощность вашей карты, проверьте один из этих веб-сайтов:

• https://developer.nvidia.com/cuda-gpus;
• https://en.wikipedia.org/wiki/CUDA#GPUs_supported.

Вы также можете использовать инструмент DIGITS device_query, чтобы проверить основные и дополнительные версии вычислений:

$ digits/device_query.py
Device #0:
>>> CUDA attributes:
  name                         Tesla K40c
  totalGlobalMem               12079136768
  clockRate                    745000
  major                        3
  minor                        5
>>> NVML attributes:
  Total memory                 11519 MB
  Used memory                  23 MB
  Memory utilization           0%
  GPU utilization              0%
  Temperature                  30 C

Драйвер

В Ubuntu вы можете установить драйвер двумя способами: с помощью файла запуска или с помощью пакета Deb.

Рекомендуется использовать пакет Deb для установки вашего драйвера, если только у вас нет нового графического процессора, который требует более новой версии драйвера. Пакеты Deb проще устанавливать, удалять и обновлять, в то время как установщики файлов запуска полезны, если вам нужна более новая версия драйвера.

Чтобы установить с помощью файла запуска, загрузите его с веб-сайта NVIDIA Driver Downloads и следуйте инструкциям. Если у вас возникнут проблемы, посмотрите раздел «Дополнительная информация».

ВАЖНО: Если вы используете файл запуска для установки своего драйвера, не устанавливайте пакет cuda Deb. Дополнительная информация ниже.

CUDA Toolkit

На веб-сайте CUDA Downloads вы увидите три варианта установки инструментария: runfile (local), deb (local), deb (network).

1. deb (network) — Это пакет Deb, предпочтительный метод. Он даёт вам доступ ко всем пакетам в репозитории CUDA (http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/), включая несколько версий инструментария.

   dpkg -i cuda-repo-ubuntu1404_7.5-18_amd64.deb
   apt-get update
   # Don't run this command yet - read below first
   #    apt-get install cuda

2. deb (local) — Также пакет Deb, хороший вариант, если у вас плохое сетевое соединение. Недостатком является то, что вы не можете получать обновления пакетов, и вам придётся устанавливать отдельные пакеты для CUDA 7.0, 7.5 и т. д.

   dpkg -i cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb
   apt-get update
   # Don't run this команду ещё - прочитайте ниже сначала
   #    apt-get install cuda

3. runfile (local) — Скрипт оболочки. Не используйте его, если в этом нет необходимости. Как и в случае с драйвером (см. выше), будет сложнее удалить или обновить вашу установку CUDA, если вы используете установщик файла запуска.

   sh cuda_7.5.18_linux.run

Пакеты Deb

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

1. apt-get install cuda — Это установит последнюю версию инструментария (в настоящее время 7.5) и последнюю версию драйвера (в настоящее время nvidia-352).
   • ВАЖНО: Не устанавливайте этот пакет, если вы установили свой драйвер с файлом запуска. Пакет Deb может не полностью удалить вашу установку драйвера с файлом запуска.
2. apt-get install cuda-toolkit-7-5 — Устанавливает только инструментарий, а не драйвер.
3. apt-get install cuda-drivers — Устанавливает только драйвер, а не инструментарий.

Для получения дополнительной информации см. раздел «Пакеты метаданных» руководства по установке CUDA для Linux. Среда будет настроена автоматически с помощью установщиков CUDA 8.0.

Наконец, вам необходимо правильно настроить среду, чтобы компоновщик времени выполнения мог найти ваши общие библиотеки. Есть несколько способов сделать это:

1. Добавьте запись в /etc/ld.so.conf.d/
   • Это требует привилегий sudo.

echo "/usr/local/cuda/lib64" | sudo tee /etc/ld.so.conf.d/cuda64.conf
sudo ldconfig

2. Отредактируйте LD_LIBRARY_PATH
   • Это не требует привилегий sudo
   • Точная формула зависит от того, какую оболочку вы используете и как вы входите на свой компьютер.

# Входная оболочка
echo "export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64" >> ~/.profile && source ~/.profile

# Интерактивная оболочка (bash)
echo "export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64" >> ~/.bashrc && source ~/.bashrc

• Дополнительная информация о настройке постоянных переменных среды:
  • http://unix.stackexchange.com/q/117467/99570
  • http://askubuntu.com/q/210884/336440

3. Установите пакет cuda-ld-conf-7-0
   • Этот пакет доступен в репозитории машинного обучения NVIDIA (http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1404/x86_64).
   • Когда вы устанавливаете DIGITS с пакетом Deb, этот пакет устанавливается автоматически, поэтому вам не нужно беспокоиться о настройке среды.
   • Он просто автоматически настраивает вариант (1) для вас.