La puissance de calcul du RTX 3090 est de 8,6 et la version de pytorch doit être supérieure à 11,1, sinon le code ne peut pas être exécuté sur le GPU et la version cuda correspondante doit également être mise à jour
Premièrement, désinstallez le cuda d'origine
sudo /usr/local/cuda-10.0/bin/uninstall_cuda_10.0.pl
sudo rm -rf /usr/local/cuda-10.0
Deuxièmement, installez cuda
cuda11.1 URL : Boîte à outils CUDA 11.1.0 | Développeur NVIDIA
Après la sélection, copiez le lien ci-dessous sur le terminal Linux
sudo sh cuda_11.1.0_455.23.05_linux.run
Pendant le processus d'installation, l'interface suivante apparaîtra
Puisque le pilote de la carte graphique a déjà été installé, sélectionnez le pilote ici, cliquez sur Entrée pour supprimer le numéro X, puis sélectionnez Installer et appuyez sur Entrée pour installer normalement. 4. Ouvrez le fichier .bashrc (dans le répertoire personnel) et entrez les lignes suivantes
exporter LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
exporter CUDA_HOME=$CUDA_HOME:/usr/local/cuda
Après avoir enregistré, quittez et entrez la source ~/.bashrc pour l'activer
Entrez nvcc -V dans le terminal pour voir si cuda est installé avec succès
Re-télécharger le pytorch correspondant à la version cuda11.1
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch-lts -c nvidia