Shortcuts

torch.accelerator.device_count

torch.accelerator.device_count()[source]

Return the number of current accelerator available.

Returns

the number of the current accelerator available.

If there is no available accelerators, return 0.

Return type

int

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources