lightning/requirements.txt

9 lines
166 B
Plaintext

scikit-learn==0.20.2
tqdm==4.35.0
numpy==1.16.4
torch>=1.2.0
torchvision>=0.3.0
pandas>=0.20.3
test-tube>=0.6.9
# future>=0.17.1 # required for buildins in setup.py