vak.common.device.get_default

vak.common.device.get_default()

get default device for torch.

Returns:

device – ‘cuda’ if torch.cuda.is_available() is True, and returns ‘cpu’ otherwise.

Return type:

str