I checked all methods in here https://pytorch.org/docs/stable/cuda.html#module-torch.cuda 16 and could not find single method that can bring me the GPU size of the device.
How to get total memory of GPU device and total available GPU memory on device?