Hi,
For me, Torch.Profiler is not working with CUDA activity only. With CPU it is working for me.
Code:
with torch.profiler.profile(
        activities=[torch.profiler.ProfilerActivity.CUDA],
        schedule=torch.profiler.schedule(wait=1, warmup=1, active=3, repeat=2),        
        on_trace_ready=torch.profiler.tensorboard_trace_handler('./log/model'),
        profile_memory=False,
        record_shapes=True,
        with_stack=True,
        use_cuda=True
    ) as prof:

        for mini_batch_data in dataloader:
              ...

Error Msg: “AssertionError: No activities specified for Kineto profiler”
Is someone able to give me an hint?
Thanks in advance!