lightning/requirements/pytorch/extra.txt

11 lines
552 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
# extended list of package dependencies to reach full functionality
matplotlib>3.1, <3.7.3
omegaconf >=2.0.5, <2.4.0
hydra-core >=1.0.5, <1.4.0
jsonargparse[signatures] >=4.18.0, <4.24.0 # strict
rich >=12.3.0, <=13.4.2
tensorboardX >=2.2, <=2.6.2 # min version is set by torch.onnx missing attribute