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

1. Visit the w.ai dashboard.

2. Login or create a w.ai account.

3. Select API Keys and create a new key.

You now have 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