numpy==1.21.2 scikit-learn==1.0.2 scipy==1.7.3 pandas==1.4.1 scikit-learn pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit==11.3.1 wandb tensorboard tensorboardX grpcio grpcio-tools protobuf==3.19.4 setuptools==61.2.0 fvcore pympler iopath