torch.cuda.get_device_properties#

torch.cuda.get_device_properties(device=None)[source]#

Get the properties of a device.

Parameters: device (torch.device or int or str, optional) – device for which to return the properties of the device. It uses the current device, given by current_device(), if device is None (default).
Returns: the properties of the device
Return type: _CudaDeviceProperties

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Find development resources and get your questions answered

View Resources