En este instructivo, se muestra cómo implementar y entregar el modelo de lenguaje grande (LLM) Llama 4 con GPUs en Google Kubernetes Engine (GKE) con el framework de entrega vLLM. Esto proporciona una base para comprender y explorar la implementación práctica de LLM para la inferencia en un entorno de Kubernetes administrado. Implementas en GKE un contenedor compilado previamente que ejecuta vLLM. También configurarás GKE para cargar Llama desde Hugging Face.
Este instructivo está dirigido a ingenieros de aprendizaje automático (AA), administradores y operadores de plataformas, y especialistas en datos y en IA que estén interesados en usar las capacidades de organización de contenedores de Kubernetes para entregar cargas de trabajo de IA/AA en hardware de GPU H200, H100, A100 y L4. Para obtener más información sobre los roles comunes y las tareas de ejemplo a las que hacemos referencia en el contenido de Google Cloud , consulta Tareas y roles comunes de los usuarios de GKE Enterprise.
Si necesitas una plataforma de IA administrada unificada diseñada para compilar y entregar modelos de AA con rapidez de forma rentable, te recomendamos que pruebes nuestra solución de implementación de Vertex AI.
Antes de leer esta página, asegúrate de estar familiarizado con los siguientes temas:
Fondo
En esta sección, se describen las tecnologías clave que se usan en este instructivo.
Llama
Llama es un modelo de lenguaje grande de Meta diseñado para una variedad de tareas de procesamiento de lenguaje natural, como la generación de texto, la traducción y la respuesta de preguntas. GKE ofrece la infraestructura necesaria para admitir las necesidades de entrenamiento y entrega distribuidos de modelos de esta escala. Para obtener más información, consulta la documentación de Llama.
GPU
Las GPUs te permiten acelerar cargas de trabajo específicas que se ejecutan en tus nodos, como el aprendizaje automático y el procesamiento de datos. GKE proporciona una variedad de opciones de tipos de máquinas para la configuración de nodos, incluidos los tipos de máquinas con GPUs NVIDIA H200, H100, L4 y A100.
vLLM
vLLM es un framework de entrega de LLM de código abierto altamente optimizado que puede aumentar la capacidad de procesamiento de entrega en GPUs, con funciones como las siguientes:
- Implementación optimizada de transformadores con PagedAttention
- Agrupación en lotes continua para mejorar la capacidad de procesamiento general de la entrega
- Paralelismo de tensor y entrega distribuida en varias GPUs
Para obtener más información, consulta la documentación de vLLM.
Objetivos
- Preparar tu entorno con un clúster de GKE en modo Autopilot o Standard.
- Implementar un contenedor de vLLM en tu clúster.
- Usa vLLM para entregar modelos de Llama 4 a través de curl y una interfaz de chat web.
Antes de comenzar
- Sign in to your Google Cloud account. If you're new to Google Cloud, create an account to evaluate how our products perform in real-world scenarios. New customers also get $300 in free credits to run, test, and deploy workloads.
-
In the Google Cloud console, on the project selector page, select or create a Google Cloud project.
-
Make sure that billing is enabled for your Google Cloud project.
-
Enable the required API.
-
In the Google Cloud console, on the project selector page, select or create a Google Cloud project.
-
Make sure that billing is enabled for your Google Cloud project.
-
Enable the required API.
-
Make sure that you have the following role or roles on the project: roles/container.admin, roles/iam.serviceAccountAdmin
Check for the roles
-
In the Google Cloud console, go to the IAM page.
Go to IAM - Select the project.
-
In the Principal column, find all rows that identify you or a group that you're included in. To learn which groups you're included in, contact your administrator.
- For all rows that specify or include you, check the Role column to see whether the list of roles includes the required roles.
Grant the roles
-
In the Google Cloud console, go to the IAM page.
Ir a IAM - Selecciona el proyecto.
- Haz clic en Grant access.
-
En el campo Principales nuevas, ingresa tu identificador de usuario. Esta suele ser la dirección de correo electrónico de una Cuenta de Google.
- En la lista Seleccionar un rol, elige un rol.
- Para otorgar funciones adicionales, haz clic en Agregar otro rol y agrega cada rol adicional.
- Haz clic en Guardar.
-
- Crea una cuenta de Hugging Face, si todavía no la tienes.
- Asegúrate de que tu proyecto tenga suficiente cuota para las GPU H100 y H200. Para obtener más información, consulta Planifica la cuota de GPU y Cuota de GPU.
Obtén acceso al modelo
Para acceder al modelo a través de Hugging Face, necesitas un token de Hugging Face.
Sigue estos pasos para generar un token nuevo si aún no tienes uno:
- Haz clic en Tu perfil > Configuración > Tokens de acceso.
- Selecciona Token nuevo.
- Especifica el nombre que desees y un rol de al menos
Read
. - Selecciona Genera un token.
- Copia el token generado al portapapeles.
Prepara el entorno
En este instructivo, usarás Cloud Shell para administrar recursos alojados enGoogle Cloud. Cloud Shell tiene preinstalado el software que necesitas para este instructivo, incluidos kubectl
y la CLI de gcloud.
Para configurar tu entorno con Cloud Shell, sigue estos pasos:
En la Google Cloud consola, haz clic en
Activar Cloud Shell en la Google Cloud consola para iniciar una sesión de Cloud Shell. Esto inicia una sesión en el panel inferior de la consola de Google Cloud .
Configura las variables de entorno predeterminadas:
gcloud config set project PROJECT_ID gcloud config set billing/quota_project PROJECT_ID export PROJECT_ID=$(gcloud config get project) export REGION=REGION export CLUSTER_NAME=CLUSTER_NAME export HF_TOKEN=HF_TOKEN
Reemplaza los siguientes valores:
PROJECT_ID
: Tu Google Cloud ID del proyecto.REGION
: una región que admita el tipo de acelerador que deseas usar, por ejemplo,us-central1
para la GPU L4.CLUSTER_NAME
: El nombre de tu clúster.HF_TOKEN
: El token de Hugging Face que generaste antes.
Crea y configura recursos de Google Cloud
Sigue estas instrucciones para crear los recursos necesarios.
Crea un clúster de GKE y un grupo de nodos
Puedes entregar modelos de Llama 4 en GPUs en un clúster de GKE Autopilot o Standard. Te recomendamos que uses un clúster de Autopilot para una experiencia de Kubernetes completamente administrada. Para elegir el modo de operación de GKE que se adapte mejor a tus cargas de trabajo, consulta Elige un modo de operación de GKE.
Autopilot
En Cloud Shell, ejecute el siguiente comando:
gcloud container clusters create-auto CLUSTER_NAME \
--project=PROJECT_ID \
--region=REGION \
--release-channel=rapid
Reemplaza los siguientes valores:
PROJECT_ID
: Tu Google Cloud ID del proyecto.REGION
: una región que admita el tipo de acelerador que deseas usar, por ejemplo,us-central1
para la GPU L4.CLUSTER_NAME
: El nombre de tu clúster.
GKE crea un clúster de Autopilot con nodos de CPU y GPU según lo solicitan las cargas de trabajo implementadas.
Estándar
En Cloud Shell, ejecuta el siguiente comando para crear un clúster estándar:
gcloud container clusters create CLUSTER_NAME \ --project=PROJECT_ID \ --region=REGION \ --workload-pool=PROJECT_ID.svc.id.goog \ --release-channel=rapid \ --num-nodes=1
Reemplaza los siguientes valores:
PROJECT_ID
: Tu Google Cloud ID del proyecto.REGION
: Una región que admita el tipo de acelerador que deseas usar, por ejemplo,us-central1
para la GPU H100.CLUSTER_NAME
: El nombre de tu clúster.
La creación del clúster puede tomar varios minutos.
Para crear un grupo de nodos para tu clúster con el tamaño de disco adecuado, ejecuta el siguiente comando:
gcloud container node-pools create gpupool \ --accelerator type=nvidia-h100-80gb,count=1,gpu-driver-version=latest \ --project=PROJECT_ID \ --location=REGION \ --node-locations=REGION-a \ --cluster=CLUSTER_NAME \ --machine-type=a3-highgpu-8g \ --disk-type=pd-ssd \ --num-nodes=1 \ --disk-size=256
GKE crea un grupo de nodos único que contiene ocho GPUs H100 de 80 GB.
Crea un secreto de Kubernetes para las credenciales de Hugging Face
En Cloud Shell, haz lo siguiente:
Configura
kubectl
para que pueda comunicarse con tu clúster:gcloud container clusters get-credentials CLUSTER_NAME \ --location=REGION
Reemplaza los siguientes valores:
REGION
: una región que admita el tipo de acelerador que deseas usar, por ejemplo,us-central1
para la GPU L4.CLUSTER_NAME
: El nombre de tu clúster.
Crea un Secret de Kubernetes que contenga el token de Hugging Face:
kubectl create secret generic hf-secret \ --from-literal=hf_api_token=${HF_TOKEN} \ --dry-run=client -o yaml | kubectl apply -f -
Reemplaza
HF_TOKEN
por el token de Hugging Face que generaste antes.
Implementa vLLM
En esta sección, implementarás el contenedor de vLLM para entregar los modelos de Llama 2 que deseas usar:
- Llama 4 Maverick 17B-128E
- Llama 4 Scout 17B-16E
Para implementar el modelo, en este instructivo se usan implementaciones de Kubernetes. Un Deployment es un objeto de la API de Kubernetes que te permite ejecutar varias réplicas de Pods que se distribuyen entre los nodos de un clúster.
Llama 4 Maverick 17B-128e
Para implementar el modelo Llama 4 Maverick 17B-128E, sigue estas instrucciones:
Crea el siguiente manifiesto
vllm-llama4-maverick-17b-128e.yaml
:Aplica el manifiesto
kubectl apply -f vllm-llama4-maverick-17b-128e.yaml
En nuestro ejemplo, limitamos la ventana de contexto a 128 K con la opción
--max-model-len=131072
de vLLM.
Llama 4 Maverick 17B-128e-it
Para implementar el modelo ajustado por instrucciones Llama 4 Maverick 17B-128e, sigue estas instrucciones:
Crea el siguiente manifiesto
vllm-llama4-maverick-17b-128e-instruct.yaml
:Aplica el manifiesto
kubectl apply -f vllm-llama4-maverick-17b-128e-instruct.yaml
En nuestro ejemplo, limitamos la ventana de contexto a 128 K con la opción
--max-model-len=131072
de vLLM.
Llama 4 Maverick 17B-128e-it-fp8
Para implementar el modelo Llama 4 Maverick 17B-128e-Instruct-FP8, sigue estas instrucciones:
Crea el siguiente manifiesto
vllm-llama4-maverick-17b-128e-instruct-fp8.yaml
:Aplica el manifiesto
kubectl apply -f vllm-llama4-maverick-17b-128e-instruct-fp8.yaml
En nuestro ejemplo, limitamos la ventana de contexto a 512 K con la opción
--max-model-len=524288
de vLLM.
Llama 4 Scout 17B-16e
Para implementar el modelo Llama 4 Scout 17B-16E, sigue estas instrucciones:
Crea el siguiente manifiesto
vllm-llama4-scout-17b-16e.yaml
:Aplica el manifiesto
kubectl apply -f vllm-llama4-scout-17b-16e.yaml
En nuestro ejemplo, limitamos la ventana de contexto a 256 K con la opción
--max-model-len=262144
de vLLM.
Llama 4 Scout 17B-16e-it
Para implementar el modelo ajustado por instrucciones Llama 4 Scout 17B-16e Instruct, sigue estas instrucciones:
Crea el siguiente manifiesto
vllm-llama4-scout-17b-16e-instruct.yaml
:Aplica el manifiesto
kubectl apply -f vllm-llama4-scout-17b-16e-instruct.yaml
En nuestro ejemplo, limitamos la ventana de contexto a 1280 K con la opción
--max-model-len=1310720
de vLLM.
Espera a que la implementación esté disponible:
kubectl wait --for=condition=Available --timeout=1800s deployment/llama-deployment
Observa los registros de la implementación en ejecución:
kubectl logs -f -l app=llama-server
El recurso de implementación descarga los datos del modelo. Este proceso puede demorar unos minutos. El resultado es similar a este:
INFO: Started server process [145]
INFO: Waiting for application startup.
INFO: Application startup complete.
...
INFO 04-07 13:36:29 [async_llm.py:228] Added request chatcmpl-4149ea4cf35e48559f9f819dcdbbb23e.
INFO: 127.0.0.1:44018 - "POST /v1/chat/completions HTTP/1.1" 200 OK
Asegúrate de que el modelo se haya descargado por completo antes de continuar con la siguiente sección.
Entrega el modelo
En esta sección, interactuarás con el modelo.
Configura la redirección de puertos
Para configurar la redirección de puertos al modelo, ejecuta el siguiente comando:
kubectl port-forward service/llama-service 8080:8000
El resultado es similar a este:
Forwarding from 127.0.0.1:8080 -> 7080
Interactúa con el modelo con curl
En esta sección, se muestra cómo puedes realizar una prueba de humo básica para verificar el modelo de Llama ajustado por instrucciones que implementaste.
Para otros modelos, reemplaza meta-llama/Llama-4-Scout-17B-16E
por el nombre del modelo respectivo.
En este ejemplo, se muestra cómo probar el modelo Llama 4 Scout 17B-16E con entrada solo de texto.
En una sesión de terminal nueva, usa curl
para chatear con tu modelo:
curl http://localhost:8080/v1/chat/completions \
-H "Content-Type: application/json" \
-d '{
"model": "meta-llama/Llama-4-Scout-17B-16E",
"messages": [{"role": "user", "content": "San Francisco is a"}],
"max_tokens": 7,
"temperature": 0
}'
El resultado es similar a este:
"message":{"role":"assistant","reasoning_content":null,"content":"San Francisco is a city","tool_calls":[]}
Soluciona problemas
- Si recibes el mensaje
Empty reply from server
, es posible que el contenedor no haya terminado de descargar los datos del modelo. Vuelve a verificar los registros del Pod en busca del mensajeConnected
, que indica que el modelo está listo para entregar. - Si ves el mensaje
Connection refused
, verifica que tu redirección de puertos esté activa.
Observa el rendimiento del modelo
Para observar el rendimiento del modelo, puedes usar la integración del panel de vLLM en Cloud Monitoring. Con este panel, puedes ver métricas de rendimiento críticas, como el rendimiento de tokens, la latencia de solicitudes y las tasas de error.
Para usar el panel de vLLM, debes habilitar Google Cloud Managed Service para Prometheus, que recopila las métricas de vLLM, en tu clúster de GKE. vLLM expone las métricas en formato de Prometheus de forma predeterminada, por lo que no necesitas instalar un exportador adicional.
Luego, puedes ver las métricas con el panel de vLLM. Si deseas obtener información para usar Google Cloud Managed Service para Prometheus y recopilar métricas de tu modelo, consulta la guía de observabilidad de vLLM en la documentación de Cloud Monitoring.Realiza una limpieza
Para evitar que se apliquen cargos a tu cuenta de Google Cloud por los recursos usados en este instructivo, borra el proyecto que contiene los recursos o conserva el proyecto y borra los recursos individuales.
Borra los recursos implementados
Para evitar que se apliquen cargos a tu cuenta de Google Cloud por los recursos que creaste en esta guía, ejecuta el siguiente comando:
gcloud container clusters delete CLUSTER_NAME \
--region=REGION
Reemplaza los siguientes valores:
REGION
: una región que admita el tipo de acelerador que deseas usar, por ejemplo,us-central1
para la GPU L4.CLUSTER_NAME
: El nombre de tu clúster.
¿Qué sigue?
- Obtén más información sobre las GPUs en GKE.
- Para aprender a usar Llama con vLLM en otros aceleradores, incluidas las GPU A100 y H100, visualiza el código de muestra en GitHub.
- Aprende a implementar cargas de trabajo de GPUs en Autopilot.
- Aprende a implementar cargas de trabajo de GPUs en Standard.
- Explora el repositorio de GitHub y la documentación de vLLM.
- Explora Vertex AI Model Garden.
- Descubre cómo ejecutar cargas de trabajo de IA/AA optimizadas con las capacidades de organización de la plataforma de GKE.