lightning/requirements/pytorch/base.txt

14 lines
567 B
Plaintext

# NOTE: the upper bound for the package version is only set for CI stability, and it is dropped while installing this package
# in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
numpy>=1.17.2, <1.23.1
torch>=1.9.*, <=1.12.0
tqdm>=4.57.0, <4.65.0
PyYAML>=5.4, <=6.0
fsspec[http]>=2021.05.0, !=2021.06.0, <2022.6.0
tensorboard>=2.9.1, <2.10.0
torchmetrics>=0.7.0, <0.9.3 # needed for using fixed compare_version
pyDeprecate>=0.3.1, <=0.3.2
packaging>=17.0, <=21.3
typing-extensions>=4.0.0, <4.3.1