Table of Contents
Shortcuts

CPUAccelerator

class pytorch_lightning.accelerators.CPUAccelerator[source]

Bases: pytorch_lightning.accelerators.accelerator.Accelerator

Accelerator for CPU devices.

static auto_device_count()[source]

Get the devices when set to auto.

Return type

int

get_device_stats(device)[source]

Get CPU stats from psutil package.

Return type

Dict[str, Any]

static get_parallel_devices(devices)[source]

Gets parallel devices for the Accelerator.

Return type

List[device]

static is_available()[source]

CPU is always available for execution.

Return type

bool

static parse_devices(devices)[source]

Accelerator device parsing logic.

Return type

int

setup_device(device)[source]
Raises

MisconfigurationException – If the selected device is not CPU.

Return type

None

teardown()[source]

Clean up any state created by the accelerator.

Return type

None

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.