Since I am new to pytorch and python. I failed to generate the “Tensor Board”
Instead, I have obtain some profile using the following code:
with torch.autograd.profiler.profile(use_cuda=True) as prof:
    net(X_Train)
print(prof)

The result is as follows:

Profile1651×722 41 KB


Trace1920×352 26.7 KB

I can see the most of the time of CUDA is spent at “aten::_fused_dropout” stage, and the total time of CUDA is 3.537ms, but it takes 98.682 second to finish the model. Ans usage of GPU is almost 0% during calculating…
Is it possible to check whether GPU is applied properly?