pytorch
torchvision
matplotlib
tensorboard
progress
torchtoolbox
numpy==1.20
spikingjelly