lightning/requirements.txt

# the default package dependencies

numpy>=1.17.2
torch>=1.4
future>=0.17.1  # required for builtins in setup.py
tqdm>=4.41.0
PyYAML>=5.1,<=5.4.1
fsspec[http]>=2021.05.0, !=2021.06.0
tensorboard>=2.2.0, !=2.5.0  # 2.5.0 GPU CI error: 'Couldn't build proto file into descriptor pool!'
torchmetrics>=0.2.0
pyDeprecate==0.3.0
packaging
typing-extensions  # TypedDict support for python<3.8
CI: split tests-examples (#990) * CI: split tests-examples * tests without template * comment depends * CircleCI typo * add doctest * update test req. * CI tests * setup macOS * longer train * lover pred acc * fix model * rename default model * lower tests acc * typo * imports * fix test optimizer * update calls * fix Win * lower Drone image * fix call * pytorch image * fix test * add dev image * add dev image * update image * drone volume * lint * update test notes * rename tests/models >> tests/base * group models * conftest * optim imports * typos * fix import * fix tests * install AMP * tests * fix import 2020-03-25 11:46:27 +00:00			`# the default package dependencies`

Auto-set `DataLoader.worker_init_fn` with `seed_everything` (#6960) Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com> Co-authored-by: ananthsub <ananth.subramaniam@gmail.com> 2021-04-19 14:28:37 +00:00			`numpy>=1.17.2`
set minimal req. PT 1.4 (#5418) * set minimal req. PT 1.4 * chlog 2021-01-13 00:15:35 +00:00			`torch>=1.4`
Removed dependency on pandas, instead use generic csv (#736) * removed dependency on pandas, instead use generic csv * remove mnist files, pushed by accident * added docstring and small fixes * Update memory.py * fixed path Co-authored-by: William Falcon <waf2107@columbia.edu> 2020-01-29 19:52:23 +00:00			`future>=0.17.1 # required for builtins in setup.py`
Warn user when IterableDataset has __len__ defined (#2437) * add warning when getting checking len * added test * changelog * pep * do not show warning below 1.4 * try version parse * comments * xfail * Update requirements/base.txt Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com> * Update pytorch_lightning/trainer/data_loading.py Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com> * version Co-authored-by: William Falcon <waf2107@columbia.edu> Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com> Co-authored-by: Jirka <jirka@pytorchlightning.ai> 2020-07-01 11:53:19 +00:00			`tqdm>=4.41.0`
Fix Namespace loading in PyYAML 5.4.x (#6673) * Fix Namespace loading in PyYAML 5.4.x * Remove OmegaConf reference from PyYAML requirements * Max allowed version for pyyaml Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> 2021-05-04 22:56:11 +00:00			`PyYAML>=5.1,<=5.4.1`
update fsspec to 2021.06.0 (#7869) 2021-06-07 23:35:19 +00:00			`fsspec[http]>=2021.05.0, !=2021.06.0`
Ban `tensorboard==2.5.0` and `deepspeed==0.3.15` (#7159) * ban TB 2.5 * note * push * Ban tb==2.5.0 and deepspeed==0.3.15 * Fix pip command * pull * up * up Co-authored-by: Carlos Mocholi <carlossmocholi@gmail.com> 2021-04-22 15:08:21 +00:00			`tensorboard>=2.2.0, !=2.5.0 # 2.5.0 GPU CI error: 'Couldn't build proto file into descriptor pool!'`
Fix `torchmetrics` compatibility (#7131) * get_num_classes * tmp * fix one test * fix deprecated tests * fix deprecate * pep8 * deprecate 0.3 * wip * wip * HaCK * brnch * brnch * format * Apply suggestions from code review * prune * rev * mltilabel * Apply suggestions from code review * master * rev * . Co-authored-by: Nicki Skafte <skaftenicki@gmail.com> 2021-04-22 20:45:46 +00:00			`torchmetrics>=0.2.0`
			`pyDeprecate==0.3.0`
fix version + yapf (#6999) 2021-04-22 18:25:51 +00:00			`packaging`
Add typing to `ModelPruning` callback (#7529) Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> 2021-05-19 20:01:42 +00:00			`typing-extensions # TypedDict support for python<3.8`