Compartir a través de

Instalación y configuración de PyTorch

En la fase anterior de este tutorial, se describen los conceptos básicos de PyTorch y los requisitos previos de su uso para crear un modelo de aprendizaje automático. Aquí, lo instalaremos en la máquina.

Consigue PyTorch

En primer lugar, deberá configurar un entorno de Python.

Se recomienda configurar un entorno de Python virtual dentro de Windows mediante Anaconda como administrador de paquetes. El resto de esta configuración supone que usa un entorno de Anaconda.

  1. Descargue e instale Anaconda aquí. Seleccione Anaconda 64-bit installer for Windows Python 3.8.

Importante

Tenga en cuenta la instalación de Python 3.x. Actualmente, PyTorch en Windows solo admite Python 3.x; Python 2.x no se admite.

Instaladores de Anaconda

Una vez completada la instalación, compruebe las versiones de Anaconda y Python.

  1. Abra el administrador de Anaconda a través de Start - Anaconda3 - Anaconda PowerShell Prompt y pruebe las versiones:

Para comprobar la versión de Python, ejecute el comando siguiente: python –-version

Para comprobar la versión de Anaconda, ejecute el siguiente comando: conda –-version

Símbolo de la línea de comandos de Anaconda

Ahora, puede instalar el paquete PyTorch desde archivos binarios a través de Conda.

  1. Navegue a https://pytorch.org/.

Seleccione los detalles de instalación de PyTorch pertinentes:

  • Compilación de PyTorch: estable.
  • Su sistema operativo: Windows
  • Paquete: Conda
  • Lenguaje: Python
  • Plataforma de proceso: CPU o elija la versión de Cuda. En este tutorial, entrenará el modelo de inferencia y la CPU, pero también podrá usar una GPU de Nvidia.
  1. Abra el administrador de Anaconda y ejecute el comando tal como se especificó en las instrucciones de instalación.
conda install pytorch torchvision torchaudio cpuonly -c pytorch

Instalación de Anaconda

  1. Confirme y complete la extracción de los paquetes necesarios.

Confirmación de la instalación de Anaconda

Vamos a comprobar la instalación de PyTorch mediante la ejecución de código de PyTorch de ejemplo para construir un tensor inicializado aleatoriamente.

  1. Abra el símbolo del sistema de Anaconda PowerShell y ejecute el siguiente comando.
python

A continuación, escriba el código siguiente:

import torch 

x = torch.rand(2, 3) 

print(x)

La salida debe ser un tensor aleatorio de 5x3. Los números serán diferentes, pero deben tener un aspecto similar al siguiente.

Comprobación de la funcionalidad de Pytorch

Nota:

¿Quiere saber más sobre el tema? Visite el sitio web oficial de PyTorch

Pasos siguientes

Ahora que hemos instalado PyTorch, estamos listos para configurar los datos de nuestro modelo.