Cómo instalar PyTorch en AlmaLinux

Popular:
¡MEJORA LA CONFIGURACIÓN DE TU SERVIDOR! APLICAR AVA Y LANZA CON UN 15% DE DESCUENTO
USA EL CÓDIGO PROMOCIONAL:

PyTorch es uno de los frameworks de machine learning de código abierto más populares desarrollados por el laboratorio de AI Research de Facebook. Se utiliza ampliamente para aplicaciones de deep learning, incluyendo computer vision y natural language processing. Si estás usando AlmaLinux, una distribución Linux estable y confiable basada en RHEL-based Linux distribution, esta guía te mostrará cómo instalar PyTorch de forma rápida y eficiente.

Requisitos previos

Antes de comenzar, asegúrate de que:

  • Tienes una instancia en ejecución de AlmaLinux 8 o 9

  • Tienes privilegios de root o sudo

  • Python 3.7 o superior está instalado

Paso 1: Actualiza tu sistema

Abre tu terminal y ejecuta el siguiente comando para actualizar todos los paquetes del sistema:

sudo dnf update -y

Opcionalmente, instala EPEL para acceder a paquetes adicionales:

sudo dnf install epel-release -y

Paso 2: Instala Python y pip

Comprueba si Python ya está instalado:

python3 --version

Si no está instalado, puedes instalarlo con:

sudo dnf install python3 python3-pip -y

Paso 3: Crea un entorno virtual (Opcional)

Para mantener tu entorno Python limpio y aislado, es una buena idea crear un entorno virtual:

python3 -m venv pytorch_env
source pytorch_env/bin/activate

Paso 4: Instala PyTorch usando pip

Para instalar PyTorch con soporte solo para CPU:

pip install torch torchvision torchaudio

Si quieres aceleración GPU y tienes una NVIDIA GPU compatible junto con los controladores adecuados y CUDA instalado, usa el comando apropiado. Por ejemplo, para instalar PyTorch con soporte CUDA 11.8:

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

Puedes encontrar las últimas opciones de instalación en el sitio web oficial de PyTorch:
https://pytorch.org/get-started/locally

Paso 5: Verifica la instalación

Para verificar que PyTorch se ha instalado correctamente, ejecuta:

python

Luego introduce lo siguiente en el shell de Python:

import torch
print(torch.__version__)
print("CUDA available:", torch.cuda.is_available())

Si todo funciona, verás la versión instalada de PyTorch y si CUDA está disponible.

Consejos para solucionar problemas

  • Si encuentras problemas con pip o los certificados SSL, actualiza pip:

pip install --upgrade pip
  • Si usas CUDA, asegúrate de que los controladores NVIDIA y el toolkit CUDA estén correctamente instalados y sean compatibles con la versión de PyTorch que estás instalando.

Conclusión

Has instalado PyTorch correctamente en AlmaLinux. Ya sea que estés desarrollando modelos de deep learning o ejecutando experimentos, PyTorch ofrece un framework flexible y potente para machine learning en sistemas Linux. Para configuraciones más avanzadas, considera integrarlo con Jupyter Notebook o usar contenedores Docker.