torch.cuda.device\_count ======================== .. currentmodule:: torch.cuda .. autofunction:: device_count