w.ai CLI Guide

To get started with the CLI, install it as mentioned in Quick Start

On a desktop environment

If on a device with a desktop environment/a browser, start by logging in:

wai login

And then run using

wai run

On a headless environment

If you want to run on a headless server, you need to run the above login command(in a desktop environment/somewhere with a browser), and then generate an API key using

wai key generate

You will get a key that you can set on the headless environment using

export W_AI_API_KEY=your key here

And then run normally (in the headless environment) as follows:

wai run

Headless environment key management

You can list generated API keys using

wai key list

And revoke any using

wai revoke <token>

For more info, run

wai key help

Specifying GPUs (NVIDIA only)

If you want to run with a subset of your GPUs, add a -g flag to the run command followed by a list of the GPU IDs. For example, to run on GPUs 0 and 1, the command would be:

wai run -g 0 1

If no GPU is provided, it will be ran on GPU 0