Încerc să-mi configurez mașina Lenovo Legion 5 Pro cu Ubuntu 20.04.3 LTS, astfel încât să îmi pot folosi GPU-ul (Nvidia Geforce RTX 3070 8GB) pentru învățarea automată, de exemplu cu PyTorch. Cu toate acestea, chiar și după reinstalarea driverelor NVidia de mai multe ori, se pare că încă nu este vizibil, deoarece rularea nvidia-smi oferă următoarele:
NVIDIA-SMI a eșuat deoarece nu a putut comunica cu NVIDIA
conducător auto. Asigurați-vă că cel mai recent driver NVIDIA este instalat și rulează.
M-am uitat la mai multe soluții acolo (acest, acest, acest, acest) și nimic nu funcționează până acum. Dar cred că problema este cauzată de kernel-ul pe care îl folosesc.
Alergare nvcc --version dă următoarea ieșire:
nvcc: driver de compilator NVIDIA (R) Cuda
Drepturi de autor (c) 2005-2019 NVIDIA Corporation
Construit pe Sun_Jul_28_19:07:16_PDT_2019
Instrumente de compilare Cuda, versiunea 10.1, V10.1.243
ls /usr/src | grep nvidia
iesiri nvidia-470.86
Când încerc să rulez aceste comenzi de la Aici, adică
sudo apt install dkms
sudo dkms install -m nvidia -v 470.86
Primesc următoarea eroare:
Eroare! Antetele dvs. de kernel pentru nucleul 5.16.2-051602-generic nu pot fi găsite.
Instalați pachetul linux-headers-5.16.2-051602-generic,
sau utilizați opțiunea --kernelsourcedir pentru a spune DKMS unde se află
Acum, linux-headers-5.16.2-051602-generic
nu există, dar deja am linux-headers-generic
instalat.
Aveți sugestii despre ce aș putea încerca să repar asta?
Mulțumesc mult!