NVIDIA-SMI a échoué car il n'a pas pu communiquer avec le pilote NVIDIA -- en-têtes du noyau pour 5.16.2

J'essaie de configurer ma machine Lenovo Legion 5 Pro w/ Ubuntu 20.04.3 LTS afin de pouvoir utiliser mon GPU (Nvidia Geforce RTX 3070 8GB) pour l'apprentissage automatique, par exemple avec PyTorch. Cependant, même après avoir réinstallé les pilotes NVidia plusieurs fois, il semble qu'il ne soit toujours pas visible, car l'exécution de nvidia-smi donne ce qui suit :

NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA 
driver. Make sure that the latest NVIDIA driver is installed and running.

J'ai examiné plusieurs solutions existantes ( ce , ce , ce , ce ) et rien ne fonctionne jusqu'à présent. Mais je pense que le problème est causé par le noyau que j'utilise.

Running nvcc --version donne le résultat suivant :

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:07:16_PDT_2019
Cuda compilation tools, release 10.1, V10.1.243

ls /usr/src | grep nvidia sorties nvidia-470.86

Lorsque j'essaie d'exécuter ces commandes à partir de aquí c'est-à-dire

sudo apt install dkms
sudo dkms install -m nvidia -v 470.86

Je reçois l'erreur suivante :

Error! Your kernel headers for kernel 5.16.2-051602-generic cannot be found.
Please install the linux-headers-5.16.2-051602-generic package,
or use the --kernelsourcedir option to tell DKMS where it's located

Maintenant, linux-headers-5.16.2-051602-generic n'existe pas, mais j'ai déjà linux-headers-generic installé.

Avez-vous des suggestions sur ce que je pourrais essayer pour résoudre ce problème ? Merci beaucoup !