用torch.cuda.Event() 精确记录GPU的运行时间:
start = torch.cuda.Event(enable_timing=True)
end = torch.cuda.Event(enable_timing=True)
start.record()
model.train(xxx)
end.record()
torch.cuda.synchronize()
elapsed_time = start.elapsed_time(end)
用torch.cuda.Event() 精确记录GPU的运行时间:
start = torch.cuda.Event(enable_timing=True)
end = torch.cuda.Event(enable_timing=True)
start.record()
model.train(xxx)
end.record()
torch.cuda.synchronize()
elapsed_time = start.elapsed_time(end)