Официальные пакеты кажутся устаревшими ( https://packages.ubuntu.com/search?keywords=nvidia-cuda-toolkit )но использование последней версии cuda, кажется, имеет значение при использовании нового оборудования.
Nvidia делает пакеты доступными, но нет гарантии, что они будут хорошо работать или будут интегрироваться с управлением драйверами nvidia в Ubuntu.
Есть ли какие-то усилия по исправлению ошибок. создать пакет ubuntu в PPA, чтобы нам не приходилось устанавливать вручную и решать конфликты индивидуально? (пакет справится с этим)
Откройте терминал и удалите все следы NVIDIA, которые могут быть в вашей системе.
sudo rm /etc/apt/sources.list.d/cuda*
sudo apt удалить --autoremove nvidia-cuda-toolkit
sudo apt удалить --autoremove nvidia-*
Установите правильный CUDA PPA в вашей системе.
экспорт ДИСТРО=ubuntu2004
экспорт АРХИТЕКТУРЫ = x86_64
судо подходящее обновление
sudo add-apt-repository ppa:графические драйверы
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/${DISTRO}/${ARCHITECTURE}/7fa2af80.pub
sudo bash -c "echo deb\ http://developer.download.nvidia.com/compute/cuda/repos/${DISTRO}/${ARCHITECTURE}/\ / > /etc/apt/sources.list.d/ cuda.list"
sudo bash -c "echo deb\ http://developer.download.nvidia.com/compute/machine-learning/repos/${DISTRO}/${ARCHITECTURE}\ / > /etc/apt/sources.list.d /cuda_learn.list"
Установите пакеты CUDA 11.1.
обновление sudo apt
sudo apt -y установить cuda-11-1
sudo apt установить libcudnn8
Укажите ПУТЬ к CUDA в файле .profile. Откройте файл, запустив sudo nano ~/.profile
и добавьте следующие строки в конец файла:
# установите PATH для установки cuda 11.1.
если [ -d "/usr/local/cuda-11.1/bin/" ]; тогда
экспорт PATH=/usr/local/cuda-11.1/bin${ПУТЬ:+:${ПУТЬ}}
экспортировать LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
фи
Используйте комбинацию клавиш Ctrl + O, а затем нажмите Enter, чтобы сохранить файл в текущем местоположении. Используйте комбинацию клавиш Ctrl + X для выхода из nano.
Перезапустите с помощью sudo reboot
и проверьте версии установки.
CUDA:
nvcc --версия
Драйвер NVIDIA:
nvidia-smi
libcudnn:
/sbin/ldconfig -N -v $(sed ‘s/:/ /’ <<< $LD_LIBRARY_PATH) 2>/dev/null | grep libcudnn