Instale anaconda para crear un entorno virtual de Python
No me centraré en cómo instalar anconda aquí. Aquí está el enlace para su referencia:
A continuación, el entorno anconda se creó correctamente. El siguiente paso es crear un entorno Python e instalar pytorch.
paso:
conda create -n nombre del entorno virtual python=número de versión
Tome python=3.7 como ejemplo para crear un entorno virtual llamado deblur.
conda crear -n deblur python = 3.7
Una vez creado el entorno de Python, instale pytorch.
Antes de instalar pytorch, debe verificar el número de versión de cuda en el servidor, porque las versiones de pytorch y cuda deben corresponder:
Instalación de la versión pytorch-cuda:
Descargue la versión del sitio web oficial de pytorch
, seleccione la adaptación correspondiente para usar diferentes instrucciones.
Utilice nvcc -V para ver el número de versión de cuda.
nvcc -V
Debe usar nvcc -V para verificar que el número de versión de cuda sea correcto.
Luego vaya al sitio web oficial de pytorch para verificar la versión correspondiente e instalarla. PyTorchhttps 
://pytorch.org/
Puede hacer clic en el enlace rodeado en un círculo en la imagen para ver versiones anteriores.
Tome pytorch como 1.7.1 y cuda como versión 10.1 como ejemplo:
Ejecutando una orden:
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch
Aparece el contenido que se muestra en la imagen, lo que indica que pytorch se instaló correctamente:
Ingrese al entorno de Python para ver si se puede usar pytorch cude:
pitón
importar antorcha
antorcha.cuda.is_available()
Si es cierto, significa que se puede utilizar.
Nota: Los pasos anteriores requieren una conexión a Internet para la instalación. De lo contrario, puede que no tenga éxito.