lightning/requirements.txt

8 lines
248 B
Plaintext

scikit-learn>=0.20.2
tqdm>=4.35.0
numpy>=1.16.4
torch>=1.1
torchvision>=0.4.0, < 0.5 # the 0.5. has some issues with torch JIT
pandas>=0.24 # lower version do not support py3.7
tensorboard>=1.14
future>=0.17.1 # required for builtins in setup.py