PyTorch CUDA memory summary, device ID 0 code example

Example: pytorch get gpu number

torch.cuda.device_count()