instalación y desinstalación de pytorch
Uno, instalación de pytorch
He instalado la versión gpu de tensorflow
1.1 Primero abra cmd para ver la versión de cuda instalada por usted mismo
ingresar
nvcc --version
Puedes ver que es cuda 10.0
1.2 Busque el paquete de descarga sin conexión e instálelo
Pytorch sin conexión dirección de descarga: Pytorch sin conexión descarga
Descargar de acuerdo con el sistema de windows + versión de python-versión cuda
1.3 Instalar usando pip
pip install torch-1.0.0-cp36-cp36m-win_amd64.whl
La versión de torchvision correspondiente a la antorcha 1.0.0 es 0.2.0
pip install torchvision==0.2.0
1.4 Verifique que la versión de gpu esté instalada correctamente
El procedimiento de prueba es el siguiente:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
resultado de la operación:
1.0.0
True
La versión de PyTorch es 1.0.0
True, lo que significa que la versión de GPU se instaló correctamente.
Dos, desinstalar pytorch
1. Use conda para desinstalar Pytorch
conda uninstall pytorch
conda uninstall libtorch
2. Use pip para desinstalar Pytorch
pip uninstall torch
Si se instaló pytorch con el comando conda install, use el primer método. Si se instaló con el comando pip install, use el segundo método. O pruebe ambos métodos.