В этой статье мы покажем, как установить NVIDIA CUDA и cuDNN в Kali Linux.
CUDA - это платформа и модель программирования для параллельных вычислений. Она позволяет программному обеспечению использовать графические процессоры для вычислений общего назначения.
CuDNN - это CUDA Deep Neural Network. Это библиотека для глубоких нейронных сетей с GPU-ускорением. CuDNN предоставляет высоконастроенные функции, которые часто возникают в приложениях с глубокими нейронными сетями.
CUDA и cuDNN необходимы для того, чтобы TensorFlow мог использовать NVIDIA GPU для ускорения кодов AI/ML.
Установка драйверов NVIDIA GPU в Kali Linux
Перед установкой CUDA и cuDNN в Kali Linux необходимо установить драйверы NVIDIA GPU в Kali Linux. Если вам нужна помощь в установке драйверов NVIDIA GPU в Kali Linux, прочтите эту статью.
Обновление кэша базы данных пакетов APT в Kali Linux
Чтобы обновить кэш базы данных пакетов APT в Kali Linux, выполните следующую команду:
| 1 | sudo apt update |
Установка NVIDIA CUDA в Kali Linux
Чтобы установить NVIDIA CUDA в Kali Linux, выполните следующую команду:
| 1 | sudo apt install nvidia-cuda-toolkit |
Для подтверждения установки нажмите Y, а затем нажмите <Enter>. Для завершения требуется некоторое время.
На данном этапе CUDA должна быть установлена в Kali Linux.
Как видно, CUDA 11.8 установлена на Kali Linux.
| 1 | nvcc --version |
Установка NVIDIA cuDNN в Kali Linux
Чтобы установить NVIDIA cuDNN в Kali Linux, выполните следующую команду:
| 1 | sudo apt install nvidia-cudnn |
После того как вы увидите следующее окно, нажмите <Enter>.
Выберите "I Agree" и нажмите на кнопку "OK".
Выполняется установка NVIDIA cuDNN в Kali Linux. Для завершения установки требуется некоторое время.
На этом этапе NVIDIA cuDNN должна быть установлена.
Заключение
Мы показали, как установить NVIDIA CUDA в Kali Linux. Также мы показали, как установить NVIDIA cuDNN в Kali Linux.