get the number of available GPU to trained on on pytorch code example Example: pytorch get gpu number torch.cuda.device_count()