installation et désinstallation de pytorch
Une, installation de pytorch
J'ai installé la version gpu de tensorflow
1.1 Commencez par ouvrir cmd pour voir la version de cuda installée par vous-même
Entrer
nvcc --version
Vous pouvez voir que c'est cuda 10.0
1.2 Trouvez le package de téléchargement hors ligne et installez-le
adresse de téléchargement hors ligne de pytorch : téléchargement hors ligne de pytorch
Télécharger selon le système Windows + version python-version cuda
1.3 Installer à l'aide de pip
pip install torch-1.0.0-cp36-cp36m-win_amd64.whl
La version de torchvision correspondant à torch 1.0.0 est 0.2.0
pip install torchvision==0.2.0
1.4 Vérifiez que la version gpu est installée avec succès
La procédure de test est la suivante:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
résultat de l'opération:
1.0.0
True
La version PyTorch est 1.0.0
True, ce qui signifie que la version GPU est installée avec succès!
Deux, désinstallation de pytorch
1. Utilisez conda pour désinstaller Pytorch
conda uninstall pytorch
conda uninstall libtorch
2. Utilisez pip pour désinstaller Pytorch
pip uninstall torch
Si pytorch a été installé avec la commande conda install, utilisez la première méthode. S'il a été installé avec la commande pip install, utilisez la deuxième méthode. Ou essayez les deux méthodes.