Deepin20.6 rtx3080 installer le lecteur de carte graphique 510.60.02, cuda 11.6, pytorch1.11

0 Préface

.Il y a eu une mission de détection de cibles récemment,À utiliserCUDAModèle de réseau de formation,L'équipement utilisé estRTX3080,InDeepinSous le système d'exploitation,Installé510.60.02Pilote de version,Notez simplement les étapes de fonctionnement,En cas de besoin

1 Travaux préparatoires

Beaucoup de messages sont téléchargés sur le site officiel de NVIDIA.runFormat drive to be installed manually,Mais j'ai essayé beaucoup de versions,Bien qu'il puisse être chargé,Mais il n'y a pas de démarrage après l'installation,Donc,,Au début de l'installation du pilote de carte vidéoCUDAAvant le kit,Il est recommandé de sauvegarder d'abord le système d'exploitation actuel

2 Nettoyage de l'environnement

Désinstallez le pilote NVIDIA actuellement installé

sudo apt autoremove nvidia*

Blindage de l'entraînement à source ouverte

sudo vi /etc/modprobe.d/blacklist.conf

Appuyez.iEntréeviMode d'édition pour,Ajouter les lignes suivantes,Et puisESCEn mode commandewqPour enregistrer

blacklist nouveau   
blacklist lbm-nouveau   
options nouveau modeset=0 
alias nouveau off   
alias lbm-nouveau off

Si l'écran noir de la carte est allumé , Ou il y a les avertissements suivants lors du démarrage

nvidiafb: cannot request PCI regions

Vous pouvez ajouter une ligne supplémentaire au fichier ci - dessus

blacklist nvidiafb

Une fois le fichier sauvegardé avec succès,Exécutez la commande suivante, Masque complet du module noyau

sudo update-initramfs -u

Redémarrer la machine, Utilisez la commande suivante pour vérifier , Le blindage a réussi sans sortie

lsmod | grep nouveau  

3 Installation d'un lecteur de carte vidéo

Je ne sais pas.Deepin La raison du système ou de mon Bureau ,Site officiel de NVIDIA.run Je n'ai pas réussi à installer le fichier pilote du formulaire , Installer l'écran noir de la carte de démarrage ,Et aprèsDeepin On l'a trouvé dans l'entrepôt officiel. driver…

sudo apt search nvidia-driver
...
...
glx-alternative-nvidia/Inconnu,now 1.2.0-1-1+eagle amd64 [Installé,Automatique]
  allows the selection of NVIDIA as GLX provider
libegl-nvidia0/Inconnu,now 510.60.02-1 amd64 [Installé,Automatique]
  NVIDIA binary EGL library
...
...

J'ai été ravi,Directinstall,Un Soha,Succès！

sudo apt install nvidia-driver
# Après l'exécution de la commande Vérifier le succès Si ça marche. nvidia-smi Il y aura une sortie d'information de carte vidéo 
nvidia-smi

4 InstallationPyTorch & CudaToolKit

Je l'ai installé.510.60.02La version vient avecCUDALa version est11.6,InpytorchSite officielJe vois.,Mise à jour actuellePyTorch StableLa version est1.11.0,SoutienCUDA 11.3,CUDA11 Compatibilité descendante de la série , Choisissez votre propre installation PyTorch 1.11.0C'est tout., Les commandes d'installation sont disponibles ci - dessous , Postez le terminal et lancez - le. ,（N'oubliez pas de le remplacer.condaOupip Sources nationales ,Plus vite.）

condaInstallation

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

pipInstallation

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

5 Vérifiez

Entrée du terminalpython or python3,EntréepythonEnvironnement interactif,Vérifiez.PyTorchEtCUDAVersion de,EtCUDADisponible ou non,Jusqu'ici.,Deepin20.6+RTX3080Installation d'un lecteur de carte vidéo、PyTorch、CUDATravaux achevés

>>> import torch
>>> torch.__version__
'1.11.0+cu113'
>>> torch.cuda.is_available()
True
>>> torch.version.cuda
'11.3'
>>>