This repository contains the code for Dynamic Graph Neural Networks on Hardware: Bottleneck Analysis (IISWC 2022).
Please install the profiling tool using this link: https://developer.nvidia.com/nsight-systems
Directly run the downloaded installer to install this tool on your machine.
pip install torch torchvision
pip install torch_tb_profiler
tensorboard --logdir=./log
log is a folder containing trace files. You can set it to any path containing the generated trace files.
Once you profile the models using Pytorch Profiler, you will get results as follows:
If you choose to use NVIDIA Nsight Systems, you will get similiar results below: