IntegrationsNebius AI Cloud
Clouds and KubernetesNebius AI Cloud

Polyaxon & Nebius AI Cloud

How to use Polyaxon and Nebius AI Cloud together

Polyaxon+

Nebius Managed Kubernetes provides the cluster foundation and optional managed GPU settings. Your platform team still selects the node-group shape, driver mode, storage, interconnect, identity, labels, taints, and scaling behavior that define each capacity class.

See the Nebius Managed Kubernetes for the upstream configuration and requirements.

Prerequisites

  • A Nebius Managed Kubernetes cluster with a GPU node group and reviewed NVIDIA driver installation mode.
  • The required interconnect, storage, registry, DNS, service account, labels, taints, and autoscaling configuration.
  • The cluster connected to Polyaxon by installing Polyaxon Agent (compute cluster), with namespace access and a queue or preset selecting that node group.

Connect the cluster

Follow Connect Kubernetes to Polyaxon to install Polyaxon Agent (compute cluster), configure storage and queues, and submit the shared GPU visibility example. Use the same namespace and queue that will run your workloads.

The example only checks GPU allocation and driver visibility. Validate storage, identity, and multi-node networking separately using the provider requirements below.

Provider requirements

  • Record the node-group shape, driver mode, labels, taints, storage, and interconnect as one validated capacity class.
  • Keep cluster, node-group, compute-cluster connection, queue, and preset configuration in version control.
  • Revalidate the GPU, storage, identity, DNS, and network path after node-image, driver, or cluster upgrades.
  • Run collective communication tests only on the exact node group and interfaces used by distributed workloads.

For distributed training, continue with NCCL and RCCL validation.

For help choosing a provider, see Choosing GPU infrastructure for Polyaxon.

Troubleshooting

The GPU resource is missing

Check node-group readiness, the selected driver installation mode, GPU operator or device-plugin status, and nvidia.com/gpu in node capacity.

Storage or interconnect devices are unavailable

Verify that the selected node group supports the requested feature, then inspect CSI or network components, topology constraints, mounts, and pod placement.

The connected compute cluster cannot submit or collect logs

Verify the Polyaxon Agent deployment, namespace RBAC, service-account permissions, registry access, queue routing, network policy, and control-plane connectivity.

References