lightning/requirements/pytorch/extra.txt

11 lines
542 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.6.2
omegaconf >=2.0.5, <2.4.0
hydra-core >=1.0.5, <1.4.0
jsonargparse[signatures] >=4.18.0, <4.22.0
rich >=12.3.0, <=13.0.1
tensorboardX >=2.2, <=2.5.1 # min version is set by torch.onnx missing attribute