onnxruntime/tools/ci_build/github/linux/docker/scripts/training/requirements.txt

13 lines
281 B
Text

--pre
-f https://download.pytorch.org/whl/torch_stable.html
# transformers requires scikit-learn
scikit-learn
numpy==1.21.6
transformers==v2.10.0
torch==1.10.0+cu113
torchvision==0.11.1+cu113
torchtext==0.11.0
tensorboard>=2.2.0,<2.5.0
h5py
setuptools<=59.5.0
parameterized>=0.8.1