Я пытаюсь настроить свой Lenovo Legion 5 Pro с машиной Ubuntu 20.04.3 LTS, чтобы я мог использовать свой графический процессор (Nvidia Geforce RTX 3070 8 ГБ) для машинного обучения, например, с PyTorch. Однако, даже после переустановки драйверов NVidia несколько раз, похоже, он все еще не виден, так как запуск nvidia-smi дает следующее:
Сбой NVIDIA-SMI, потому что он не может связаться с NVIDIA.
Водитель. Убедитесь, что установлена и запущена последняя версия драйвера NVIDIA.
Я просмотрел несколько решений там (это, это, это, это) и ничего не работает до сих пор. Но я думаю, что проблема вызвана ядром, которое я использую.
Бег nvcc --версия дает следующий результат:
nvcc: драйвер компилятора NVIDIA (R) Cuda
Авторское право (c) 2005-2019 Корпорация NVIDIA
Построено Sun_Jul_28_19:07:16_PDT_2019
Инструменты компиляции Cuda, выпуск 10.1, V10.1.243
лс /usr/источник | grep нвидиа
выходы nvidia-470.86
Когда я пытаюсь запустить эти команды из здесь, т.е.
sudo apt установить dkms
sudo dkms установить -m nvidia -v 470.86
Я получаю следующую ошибку:
Ошибка! Заголовки ядра для ядра 5.16.2-051602-generic не найдены.
Установите пакет linux-headers-5.16.2-051602-generic,
или используйте параметр --kernelsourcedir, чтобы сообщить DKMS, где он находится.
В настоящее время, Linux-заголовки-5.16.2-051602-общий
не существует, но у меня уже есть linux-заголовки-универсальные
установлен.
Есть ли у вас какие-либо предложения о том, что я мог бы попытаться исправить это?
Большое спасибо!