lightning/requirements/pytorch/base.txt

13 lines
543 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.24.4
torch >=1.11.0, <=2.0.0
tqdm >=4.57.0, <4.66.0
PyYAML >=5.4, <=6.0
fsspec[http] >2021.06.0, <2023.5.0
torchmetrics >=0.7.0, <0.12.0 # needed for using fixed compare_version
packaging >=17.1, <=23.0
typing-extensions >=4.0.0, <=4.4.0
lightning-utilities >=0.8.0, <0.9.0