Cómo instalar PyTorch en Windows y Linux

En esta guía, voy a ir a través de la instalación de PyTorch en Windows y Linux, utilizando Anaconda para gestionar la instalación de los paquetes.

La instalación se realizará utilizando la herramienta de línea de comandos conda incorporada en Anaconda.

PyTorch es una biblioteca de aprendizaje automático escrita en Python y basada en el marco Torch. Fue desarrollada por Facebook y es comparable a Tensorflow de Google. Es útil en los campos de la visión por ordenador y el procesamiento del lenguaje natural y ha sido utilizada por empresas como Tesla para desarrollar software de piloto automático.

PyTorch es gratuito y de código abierto, con licencia BSD modificada, y está bajo la Fundación Linux.

Requisitos previos

Para seguir este tutorial, necesita tener instalado Anaconda en la máquina en la que esté trabajando.

Si aún no lo tiene instalado, esta guía sobre cómo instalar Anaconda le guiará a través de todo el proceso. Después de seguir esa guía, puede proceder a instalar PyTorch.

Instalación de PyTorch en Linux

Como buena práctica, comience por actualizar los paquetes de software en su distribución de Linux. En mi caso, estoy utilizando Ubuntu y apt para gestionar mis paquetes, por lo que utilizaré el siguiente comando para actualizar:

$ sudo apt update && apt upgrade 

Cuando haya terminado de actualizar los paquetes, vaya a la página de instalación del sitio web oficial de PyTorch. Desplácese por la página hasta que encuentre el asistente de instalación que se parece al de abajo:

Utilizando este asistente, podrá hacer clic en las diferentes opciones para proporcionar la información de su sistema y sus preferencias, y a su vez, obtendrá un comando que puede utilizar en su terminal para instalar PyTorch.

Después de proporcionar la información de mi sistema, este es el aspecto que tiene:

Opté por la versión estable de Linux, y utilizaré Conda para gestionar mis paquetes. También elegí utilizar PyTorch con Python en lugar de C /Java. Y estaré ejecutando mi PyTorch en una CPU en lugar de una GPU.

En la parte inferior de la tabla está el comando que puedo utilizar para instalar PyTorch, pero antes de ejecutar este comando, me gustaría crear un entorno virtual Anaconda llamado pytorch.

Los entornos virtuales le permiten crear proyectos y mantener sus dependencias aisladas de las dependencias de otros proyectos, evitando así conflictos de dependencias. Una de las ventajas de Anaconda es que le ayuda a crear y gestionar entornos virtuales fácilmente.

Para crear un entorno virtual en el que la versión de Python sea la 3.7, introduciré el siguiente comando:

conda create -n pytorch python=3.7

Una vez creado el entorno, lo activaré utilizando el siguiente comando:

conda activate pytorch

Una vez que el entorno esté activo, ejecutaré el comando generado anteriormente en el sitio web de PyTorch para instalar PyTorch.

conda install pytorch torchvision torchaudio cpuonly -c pytorch

Siga las instrucciones para instalar PyTorch. Una vez hecho, reiniciaré la sesión de terminal para que surta efecto.

Ahora para verificar que PyTorch fue instalado correctamente, vamos a intentar importarlo en la shell interactiva de Python. Asegúrese de estar en el entorno virtual de Pytorch utilizando el comando

conda activate pytorch

Una vez que esté en el entorno virtual de Pytorch, abra la shell interactiva de python escribiendo el comando

python

Una vez iniciada la sesión de shell, escriba la siguiente línea de código y pulse ENTER

importar antorcha

Si Python se ejecuta sin errores, entonces la instalación se ha realizado correctamente. Pero si obtiene un error de Módulo no encontrado, significa que algo salió mal durante la instalación. Puede intentar reinstalarlo de nuevo.

Instalación de PyTorch en Windows

Para empezar, en su máquina Windows, busque el programa Anaconda Prompt y ábralo. Aquí es donde vamos a ejecutar los comandos.

Una vez que el programa esté abierto, vamos a crear un entorno virtual para nuestra instalación de PyTorch utilizando el comando

conda create -n pytorch python=3.7

Después de crear el entorno virtual, podemos activarlo ejecutando el siguiente comando:

conda activate pytorch

Una vez que el entorno virtual está activo, podemos proceder a instalar PyTorch. Comenzamos yendo a la página de instalaciones del sitio web de PyTorch. Después, podemos desplazarnos hasta la sección de la página donde se encuentra el asistente de instalación:

Aquí, seleccionamos la información de nuestro sistema, y el asistente nos dará un comando para instalar PyTorch. Voy a seleccionar la versión estable para Windows, gestionada por Conda, utilizada a través del lenguaje de programación python, y que se ejecuta en una CPU. Como resultado, mi tabla tendrá este aspecto.

A continuación, copie el comando, péguelo en el prompt de Anaconda y pulse ENTER.

Una vez completada la instalación, podemos verificar si ha tenido éxito abriendo el intérprete de comandos interactivo de Python e intentando importar PyTorch.

Así que dentro del prompt de Anaconda, inicie una sesión interactiva de Python.

python

Una vez iniciada la sesión, importe PyTorch utilizando la siguiente línea de código:

import antorcha

Si esta acción se completa sin errores, entonces la instalación fue exitosa.

Palabras finales

En esta guía, hemos instalado PyTorch tanto en Windows como en Linux utilizando conda. Es posible instalarlo vía PIP como un paquete PIP normal. En ambos casos, opté por la instalación con CPU. Sin embargo, también puede utilizar CUDA, que es un conjunto de herramientas de sistema desarrollado por Nvidia que acelera el entrenamiento paralelizando las operaciones en las GPU.