onnxruntime/tools/ci_build/github/linux/docker/scripts/training/requirements.txt
baijumeswani 79f832c682
Separate requirements.txt file for ORTModule pipelines (#6879)
* Move all ORTModule dependency installations to ortmodule subfolder
2021-03-05 14:12:11 -08:00

11 lines
222 B
Text

--pre
-f https://download.pytorch.org/whl/torch_stable.html
# transformers requires sklearn
sklearn
numpy==1.16.6
transformers==v2.10.0
torch==1.6.0+cu101
torchvision==0.7.0+cu101
torchtext==0.7.0
tensorboard==v2.0.0
h5py