1. Установка драйвера на видеокарту (если не установлен)

Скачиваем драйвер с официального сайта, выбрав модель своей видеокарты: https://www.nvidia.ru/Download/index.aspx?lang=ru

В моём случае этот файл называется NVIDIA-Linux-x86_64-440.44.run

Чтобы его установить, необходимо сначала переключиться в режим терминала Ctrl + Alt + F1, а дальше сделать следующее по списку:

  1. Отключение сервиса lightdm (графика): sudo service lightdm stop
  2. Запуск скаченного файла: sudo bash NVIDIA-Linux-x86_64-440.44.run
  3. Перезагрузка: sudo reboot

Если всё сработало, команда nvidia-smi должна вывести информацию о видеокарте и версии драйвера в следующем виде:

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