lightning/requirements/pytorch/base.txt

12 lines
407 B
Plaintext
Raw Normal View History

numpy>=1.17.2, <1.22.5
torch>=1.9.*, <=1.11.0
tqdm>=4.57.0, <=4.63.0
PyYAML>=5.4, <=6.0
fsspec[http]>=2021.05.0, !=2021.06.0, <=2022.2.0
tensorboard>=2.2.0, <2.10.0
torchmetrics>=0.4.1, <0.9.2
pyDeprecate>=0.3.1, <=0.3.2
packaging>=17.0, <=21.3
typing-extensions>=4.0.0, <4.2.1
2022-05-28 21:30:42 +00:00
protobuf<=3.20.1 # strict. TODO: Remove after tensorboard gets compatible https://github.com/tensorflow/tensorboard/issues/5708