Pytorch sitio web oficial de la dirección
En la actualidad, la última versión de Pytorch es 1.4, compatible con las últimas cuda9.2 y 10.1
según su propia versión CUDA, puede elegir la versión adecuada de Pytorch
(Debe elegir adaptarse a la versión CUDA de su propia computadora o servidor, de lo contrario, ¡cometerá un error!)
Haga clic en la letra roja para instalar versiones anteriores de PyTorch en la imagen a continuación,
que es esta URL: https://pytorch.org/get-started/previous-versions/
Consulte la siguiente interfaz:
seleccione el comando de versión correspondiente según sea necesario.
Aquí lo instalé basado en CUDA9.0 y Anaconda, por lo que el comando seleccionado es:
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0
Tenga en cuenta que "-c pytorch" se elimina aquí. Se utilizará la fuente de espejo doméstico preconfigurada y la velocidad de descarga será mucho más rápida. El método de configuración específico es muy simple, pero lo repetiré.
Una vez completada la instalación, ingrese:
python
Ingrese el entorno python de anaconda
e ingrese:
import torch
import torchvision
print(torch.__version__)
print(torchvision.__version__)
Obtenga la siguiente pantalla,
puede ingresar nuevamente:
print(torch.cuda.is_available())
Pantalla: verdadero
¡La instalación está completa!