Rate this Page
★ ★ ★ ★ ★

torch.mps.device_count#

torch.mps.device_count()[source]#

Returns the number of available MPS devices.

Return type:

int

On this page
Show Source
PyTorch Libraries