Instalación de Pytorch y configuración del entorno

https://pytorch.org/get-started/previous-versions/

https://blog.csdn.net/thomaswu1992/article/details/90293015

Use conda para instalar, primero cree el entorno virtual de pytorch pytorch-py36 antes de la instalación, python elige la versión 3.6

 

Luego active pytorch-py36 en el indicador de Anaconda para ingresar a este entorno, copie el siguiente código en él y presione Entrar para instalar.

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch

 Finalmente, en el entorno pytorch-py36, ingrese python e ingrese el siguiente código para probar la instalación

import torch   # 导入torch包 如果没有报错则表示安装成功
torch.cuda.is_available()  # 返回True则表示GPU可用， 反之则不可用

Debido a que CUDA9.0 y la versión correspondiente de cuDNN se instalaron antes de instalar tensorflow, esta instalación de Pytorch no implicó la descarga de CUDA y cuDNN del sitio web oficial de NVIDIA.

Sin embargo, durante la instalación, descargué cientos de megabytes de cudatoolkit 9.0.