¿Cómo instalar Pytorch en Ubuntu 22.04?

PyTorch es un framework de aprendizaje automático de código abierto ampliamente utilizado que destaca por su facilidad de uso, su grafo de computación dinámico y su eficiencia, lo que lo hace ideal para tareas de deep learning.

Desarrollado por Meta AI y actualmente parte de la Linux Foundation, PyTorch se utiliza en aplicaciones como reconocimiento de imágenes y voz, procesamiento del lenguaje natural y aprendizaje por refuerzo. Su potente arquitectura optimizada para GPU permite a los científicos de datos crear e implementar redes neuronales con facilidad, convirtiéndolo en una herramienta esencial para la investigación y el desarrollo en IA.

Esta guía te ayudará a instalar PyTorch en Ubuntu usando Pip o Anaconda para comenzar.

Requisitos previos

Asegúrate de que tu sistema cumpla estos requisitos:

• Python 3.6 o una versión más reciente
• Para soporte de GPU, asegúrate de que los controladores y el toolkit de CUDA estén instalados.

Instalar PyTorch en Ubuntu (Install Pytorch Ubuntu)

Para instalar PyTorch en un sistema Ubuntu, ya sea con soporte de CPU o GPU, puedes usar Pip o Anaconda. Nos centraremos en cómo instalar PyTorch usando Pip y en configurar un entorno virtual para mantener tus paquetes de Python organizados.

Método 01: Instalar PyTorch en Ubuntu usando Pip3 (Install Pytorch Ubuntu installation)

Puedes instalar PyTorch fácilmente en Ubuntu usando Pip (Python package Manager) siguiendo estos pasos:

Paso 1: Actualizar los paquetes del sistema

Primero, asegúrate de que los paquetes del sistema estén actualizados. Abre tu terminal y ejecuta el siguiente comando:

$ sudo apt update

Esto actualizará las listas de paquetes para garantizar que tengas las versiones más recientes.

Paso 2: Instalar Python3-venv

A continuación, necesitarás instalar python3-venv, que te permite crear entornos de Python aislados. Esto es importante para evitar conflictos entre proyectos que requieren distintas versiones de paquetes.

$ sudo apt install python3-venv -y

Paso 3: Configurar un entorno virtual de Python

Para configurar un entorno virtual de Python, crea un directorio para tu proyecto y utiliza el comando cd para entrar en él. Este directorio contendrá el entorno virtual y los archivos relacionados. Por ejemplo, hemos creado un directorio llamado “pytorch_env“.

$ mkdir pytorch_env
$ cd pytorch_env

Ahora, crea y activa el entorno virtual:

python3 -m venv pytorch_env
source pytorch_env/bin/activate

Una vez activado el entorno virtual, el prompt del terminal cambiará para mostrar que ahora estás trabajando dentro del entorno pytorch_env. En este punto, puedes instalar paquetes de Python o ejecutar scripts específicamente dentro de este entorno. Esto ayuda a evitar conflictos con otros proyectos al mantener las dependencias separadas de la instalación de Python a nivel del sistema.

Paso 4: Instalar PyTorch usando Pip (Install Pytorch Ubuntu using Pip)

Con el entorno virtual activo, puedes instalar PyTorch. Si no tienes GPU o no necesitas aceleración por GPU, puedes instalar la librería de PyTorch solo con soporte de CPU. Para instalar PyTorch para ejecutarse en CPU, usa este comando:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

Si tienes una GPU compatible y quieres aprovechar CUDA para aceleración, instala PyTorch con soporte de GPU:

pip3 install torch torchvision torchaudio

Paso 5: Verificar la instalación

Para confirmar que PyTorch se ha instalado correctamente, abre el intérprete de Python y comprueba la versión de PyTorch:

python
import torch

value = torch.randn(1, 6)

print(torch.version)

Esto mostrará la versión de PyTorch instalada, confirmando que la configuración fue exitosa.

Método 02: Cómo instalar PyTorch en Ubuntu usando Anaconda

Puedes instalar PyTorch en Ubuntu usando Anaconda, una plataforma open-source muy popular. Este método permite gestionar fácilmente tus paquetes y entornos de Python. Para instalar PyTorch usando Conda, sigue estos pasos:

Paso 1: Actualizar los paquetes del sistema

Comienza actualizando los paquetes del sistema para asegurarte de que todo esté al día. Abre tu terminal y ejecuta:

$ sudo apt update

Paso 2: Instalar Anaconda o Conda

A continuación, necesitas instalar Anaconda. Puedes instalar paquetes de Conda usando el comando Curl. Si curl no está instalado en tu sistema Ubuntu, instálalo con el siguiente comando:

$ sudo apt install curl -y

Navega a un directorio temporal: es una buena práctica realizar instalaciones en un directorio aislado. Puedes usar el directorio /tmp:

cd /tmp

Descarga el script instalador de Anaconda usando Curl:

curl --output anaconda.sh https://repo.anaconda.com/archive/Anaconda3-2023.07-2-Linux-x86_64.sh

Este comando guarda el instalador de Anaconda como “anaconda.sh”. Puedes consultar el sitio web oficial de Anaconda para obtener la versión más reciente si es necesario.

Verifica el archivo descargado. Es importante comprobar la integridad del archivo descargado para asegurarte de que no se haya corrompido. Verifícalo con:

sha256sum anaconda.sh

Asegúrate de que el checksum coincida con el valor proporcionado en el sitio web oficial de Anaconda.

Una vez verificado, continúa con la instalación ejecutando el script:

bash anaconda.sh

Después de la instalación, actualiza tu sesión de shell para que los comandos de Anaconda estén disponibles en tu terminal:

source ~/.bashrc

Deberías ver que el prompt del terminal cambia a “base”, lo que indica que el entorno predeterminado de Anaconda está activo. Más adelante puedes activar Anaconda usando el comando conda activate.

Confirma que Anaconda está instalado ejecutando:

conda --version

Paso 3: Instalar PyTorch usando Anaconda

Con Anaconda instalado y activo, ya puedes instalar PyTorch.

Si tu sistema no tiene GPU o no necesitas aceleración por GPU, instala PyTorch con soporte de CPU:

conda install pytorch torchvision torchaudio cpuonly -c pytorch

Si tienes GPU y quieres usar CUDA para aceleración, instala PyTorch con soporte de GPU:

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

Paso 4: Verificar la instalación

Por último, para verificar que PyTorch se instaló correctamente, inicia una sesión de Python y comprueba la versión de PyTorch:

python
import torch
print(torch.version)

Esto imprimirá la versión de PyTorch instalada actualmente, confirmando que todo está configurado correctamente.

Cómo desinstalar PyTorch en Ubuntu

Para desinstalar PyTorch de tu sistema Ubuntu, sigue estos pasos según el método con el que lo instalaste originalmente.

Si instalaste PyTorch usando Pip, puedes eliminarlo ejecutando el siguiente comando:

$ pip3 uninstall torch -y

Este comando desinstalará PyTorch de tu entorno de Python.

Si usaste Anaconda para instalar PyTorch, puedes desinstalarlo con este comando:

$ conda remove pytorch -y

Esto eliminará PyTorch de tu entorno de Anaconda.

Conclusión

Esta guía mostró cómo instalar PyTorch en Ubuntu usando tanto Pip como Anaconda. También cubrió las distintas opciones de instalación disponibles según los recursos computacionales de tu sistema. Con PyTorch ya configurado, estás listo para explorar sus funciones y aprovechar sus potentes capacidades.

El hosting web VPS es una opción potente y flexible que puede influir significativamente en el éxito de tu sitio web. Con blueVPS, puedes alquilar servidores en todo el mundo, incluyendo cPanel para una gestión sencilla. ¿Listo para mejorar el rendimiento de tu web? Considera el hosting VPS hoy mismo.