Compartir a través de

Instale y configure PyTorch en la máquina.

  • Artículo

Nota:

Para obtener una mayor funcionalidad, PyTorch también se puede usar con DirectML en Windows.

En la fase anterior de este tutorial analizamos los conceptos básicos de PyTorch y los requisitos previos para usarlo para crear un modelo de aprendizaje automático. En este caso, lo instalaremos en la máquina.

Obtener PyTorch

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

Es recomendable configurar un entorno de Python virtual dentro de Windows, mediante Anaconda como administrador de paquetes. En el resto de esta configuración se da por supuesto 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 que debe instalar Python 3.x. Actualmente, PyTorch en Windows solo admite Python 3.x; en cambio, no se admite Python 2.x.

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 Iniciar - Anaconda3 - Símbolo del sistema de Anaconda PowerShell, y pruebe las versiones:

Puede comprobar la versión de Python si ejecuta el comando siguiente: python –-version.

Puede comprobar la versión de Anaconda si ejecuta el comando siguiente: conda –-version.

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

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

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

Seleccione los detalles de instalación de PyTorch pertinentes:

  • Compilación de PyTorch: estable.
  • Su sistema operativo: Windows
  • Paquete: Conda
  • Idioma: Python
  • Plataforma del 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.

Configuración de Anaconda

  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

Aquí tendrá que comprobar la instalación de PyTorch mediante la ejecución del 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 comando siguiente:

import torch 

x = torch.rand(2, 3) 

print(x)

La salida debe ser un tensor aleatorio de 2 x 3. 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 ya ha instalado PyTorch, estará listo para configurar los datos de nuestro modelo.