tim = 0 start = torch.cuda.Event(enable_timing=True) #the times end = torch.cuda.Event(enable_timing=True) start.record() out = net(image_input) end.record() torch.cuda.synchronize() tim = start.elapsed_time(end)
pytorch计算GPU运行时间:
于 2022-02-18 16:57:31 首次发布