lightning/requirements/pytorch/base.txt

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

13 lines
534 B
Plaintext
Raw Normal View History

# 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.2
torch>=1.11.0, <=2.0.0
tqdm>=4.57.0, <4.65.0
PyYAML>=5.4, <=6.0
fsspec[http]>2021.06.0, <2023.2.0
torchmetrics>=0.7.0, <0.12.0 # needed for using fixed compare_version
2023-01-26 12:42:57 +00:00
packaging>=17.1, <=23.0
2022-10-31 13:52:45 +00:00
typing-extensions>=4.0.0, <=4.4.0
lightning-utilities>=0.8.0, <0.9.0