Table of Contents
Shortcuts

IPUAccelerator

class pytorch_lightning.accelerators.IPUAccelerator[source]

Bases: pytorch_lightning.accelerators.accelerator.Accelerator

Accelerator for IPUs.

static auto_device_count()[source]

Get the devices when set to auto.

Return type:

int

get_device_stats(device)[source]

IPU device stats aren’t supported yet.

Return type:

Dict[str, Any]

static get_parallel_devices(devices)[source]

Gets parallel devices for the Accelerator.

Return type:

List[int]

static is_available()[source]

Detect if the hardware is available.

Return type:

bool

static parse_devices(devices)[source]

Accelerator device parsing logic.

Return type:

int

To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. Read PyTorch Lightning's Privacy Policy.