1. Désinstallez le pilote d'origine
sudo apt remove *cuda*
sudo apt remove *nvidia*
sudo /usr/bin/nvidia-uninstall
sudo dpkg -l | grep ^rc | cut -d' ' -f3 | sudo xargs dpkg --purge
sudo rm -rf ~/.cuda-license-*
sudo apt purge nvidia-cuda-toolkit
sudo apt remove nvidia-driver-*
sudo apt purge nvidia-*
2. Désactivez le nouveau pilote
Modifier /etc/modprobe.d/blacklist.conf
, il est préférable d'ajouter le code suivant
blacklist nouveau
options nouveau modeset=0
Exécutez la commande sudo update-initramfs -u
après le redémarrage lsmod | grep nouveau
. Si aucun affichage ne s'affiche, la désactivation est réussie.
3. Téléchargez et installez le pilote
télécharger
sudo wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run
Installer
sudo sh cuda_11.7.0_515.43.04_linux.run
Entrez accept
pour confirmer
et sélectionnez Install
Installer
Vérifier
nvcc -V
nvidia-smi
autre
Docker exécute le runtime=nvidia spécifié et signale une erreur
unknown or invalid runtime name: nvidia
Causes possibles
Pas installénvidia-container-runtime
solution
Effectuer l'installation nvidia-container-runtime
sudo apt install nvidia-container-runtime
Modifier/etc/docker/daemon.json
{
"registry-mirrors": [
"https://docker.mirrors.ustc.edu.cn",
"https://hub-mirror.c.163.com",
"https://registry.docker-cn.com"
],
"runtimes": {
"nvidia": {
"args": [],
"path": "nvidia-container-runtime"
}
}
}
Redémarrer le menu fixe
sudo systemctl daemon-reload
sudo systemctl restart docker