import torch a = torch.cuda.is_available() b = torch.cuda.current_device() c = torch.cuda.get_device_name() d = torch.cuda.memory_reserved() e = torch.cuda.memory_al