FS-TFP/environment/requirements-torch1.10.txt

19 lines
257 B
Plaintext

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