torch.cuda.get_device_properties
torch.cuda.get_device_properties(device)[source]-
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
© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.cuda.get_device_properties.html