How to detect if the GPUs are AMD's or NVIDIA's

1

Hello,

I would like to know if there is a way to detect the type of the GPU in terms of manufacturer.

torch.cuda.get_device_capability('cuda') gives (8, 0) for NVIDIA A100 and (9,0) for AMD MI250X. It seems that the result is also (9,0) for NVIDIA H100, so I’m not sure how to distinguish between NVIDIA and AMD.

Thank you in advance for your help!