--device-arch flag of the command you run. If you need to pin a workload to a specific cluster — for instance, to target a particular accelerator model — you can pass --affinity cluster=<cluster_name> to flexai training run, flexai training debug-ssh, or flexai inference serve.
Use the flexai cluster set of subcommands to discover which clusters your organization can use.
Available subcommands
flexai cluster list- Lists the clusters available to your organization.