onnxruntime/tools/ci_build/github/linux/docker/scripts/training/requirements.txt
2021-02-27 09:35:42 -08:00

15 lines
No EOL
344 B
Text

--pre
-f https://download.pytorch.org/whl/nightly/cu101/torch_nightly.html
# transformers requires sklearn
pandas
sklearn
numpy==1.16.6
transformers==v2.10.0
torch
torchvision
torchtext
tensorboard==v2.0.0
h5py
wget
# PyTorch Lightning (nightly) is used for CI tests only
https://github.com/PyTorchLightning/pytorch-lightning/archive/master.zip