Jongbin Jung
04/27/2026, 3:52 PMMarvin
04/27/2026, 3:52 PMMarvin
04/27/2026, 3:56 PMbash
prefect work-pool create my-vertex-pool --type vertex-ai
2. Set GPU-related job_variables on the deployment
yaml
deployments:
- name: my-gpu-deployment
entrypoint: flows/my_flow.py:my_flow
work_pool:
name: my-vertex-pool
job_variables:
region: us-central1
image: <http://gcr.io/my-project/my-image:latest|gcr.io/my-project/my-image:latest>
machine_type: n1-standard-4
accelerator_type: NVIDIA_TESLA_T4
accelerator_count: 1
Important notes
- Your container image needs the right CUDA / ML libraries if your code expects GPU access.
- The exact GPU types available depend on GCP/Vertex AI regional availability.
- In Prefect 3, these GPU settings come from the prefect-gcp Vertex AI worker integration.
Relevant docs / references
- Prefect GCP Vertex AI worker API reference
- GCP Vertex AI MachineSpec reference
If you want, I can also show you:
- a full prefect.yaml example for Vertex AI with GPU, or
- how to migrate an existing Cloud Run deployment to Vertex AI.Jongbin Jung
04/27/2026, 3:57 PMJongbin Jung
04/27/2026, 3:58 PMMarvin
04/27/2026, 3:58 PMMarvin
04/27/2026, 3:58 PM