1. Установка драйвера на видеокарту (если не установлен)
Скачиваем драйвер с официального сайта, выбрав модель своей видеокарты: https://www.nvidia.ru/Download/index.aspx?lang=ru
В моём случае этот файл называется NVIDIA-Linux-x86_64-440.44.run
Чтобы его установить, необходимо сначала переключиться в режим терминала Ctrl + Alt + F1, а дальше сделать следующее по списку:
- Отключение сервиса lightdm (графика): sudo service lightdm stop
- Запуск скаченного файла: sudo bash NVIDIA-Linux-x86_64-440.44.run
- Перезагрузка: sudo reboot
Если всё сработало, команда nvidia-smi должна вывести информацию о видеокарте и версии драйвера в следующем виде:
№№ 2. Скачивание и установка CUDA.
Скачиваем также с официального сайта: https://developer.nvidia.com/cuda-downloads
Выбирает deb (local) и в появившемся окне выполняем перечисленные команды:
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-ubuntu1604.pin sudo mv cuda-ubuntu1604.pin /etc/apt/preferences.d/cuda-repository-pin-600 wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda-repo-ubuntu1604-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb sudo dpkg -i cuda-repo-ubuntu1604-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb sudo apt-key add /var/cuda-repo-10-2-local-10.2.89-440.33.01/7fa2af80.pub sudo apt-get update sudo apt-get -y install cuda
После этого перезагружаемся и пробуем nvidia-smi снова.
3. Добавление пути к CUDA.
В файл ~/.bashrc добавляем export PATH=/usr/local/cuda/bin:$PATH. Теперь в терминале nvcc --version должна отображаться версия компилятора.
nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2019 NVIDIA Corporation Built on Wed_Oct_23_19:24:38_PDT_2019 Cuda compilation tools, release 10.2, V10.2.89