---
title: Accelerator
description: The Accelerator base class for Lightning PyTorch.
canonical: >-
  https://lightning.ai/docs/pytorch/stable/optional-api/api/lightning.pytorch.accelerators.Accelerator
---
# Accelerator {#accelerator}

### `class lightning.pytorch.accelerators.Accelerator` {#lightning.pytorch.accelerators.Accelerator}

Bases: [`Accelerator`](/docs/fabric/stable/api/generated/lightning.fabric.accelerators.Accelerator#lightning.fabric.accelerators.Accelerator "(in lightning v2.7.0.dev)"), [`ABC`](https://docs.python.org/3/library/abc.html#abc.ABC "(in Python v3.15)")

The Accelerator base class for Lightning PyTorch.

Warning

Writing your own accelerator is an [experimental](/docs/pytorch/2.5.5/reference/versioning.md#experimental-api) feature.

#### `get_device_stats(device)` {#lightning.pytorch.accelerators.Accelerator.get_device_stats}

Get stats for a given device.

Parameters:

**device** (`Union`\[[`device`](https://docs.pytorch.org/docs/stable/tensor_attributes.html#torch.device "(in PyTorch v2.15)"), [`str`](https://docs.python.org/3/builtins/stdtypes.html#str "(in Python v3.15)"), [`int`](https://docs.python.org/3/builtins/functions.html#int "(in Python v3.15)")\]) – device for which to get stats

Return type:

[`dict`](https://docs.python.org/3/builtins/stdtypes.html#dict "(in Python v3.15)")\[[`str`](https://docs.python.org/3/builtins/stdtypes.html#str "(in Python v3.15)"), [`Any`](https://docs.python.org/3/library/typing.html#typing.Any "(in Python v3.15)")\]

Returns:

Dictionary of device stats

#### `setup(trainer)` {#lightning.pytorch.accelerators.Accelerator.setup}

Called by the Trainer to set up the accelerator before the model starts running on the device.

Parameters:

**trainer** ([`Trainer`](/docs/pytorch/2.5.5/optional-api/api/lightning.pytorch.trainer.trainer.Trainer.md#lightning.pytorch.trainer.trainer.Trainer "lightning.pytorch.trainer.trainer.Trainer")) – the trainer instance

Return type:

[`None`](https://docs.python.org/3/builtins/constants.html#None "(in Python v3.15)")

---

**Navigation**

- Parent: [Api](/docs/pytorch/2.5.5/optional-api/api.md)