torch.cuda.get_device_properties
torch.cuda.get_device_properties(device)
Gets the properties of a device.
Parameters:
      device (torch.device or int or str) -- device for
      which to return the properties of the device.
Returns:
      the properties of the device
Return type:
      _CudaDeviceProperties