- 博客(1)
- 收藏
- 关注
原创 pytorch计算GPU运行时间:
tim = 0start = torch.cuda.Event(enable_timing=True) #the timesend = torch.cuda.Event(enable_timing=True)start.record()out = net(image_input)end.record()torch.cuda.synchronize()tim = start.elapsed_time(end)
2022-02-18 16:57:31
1944
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人