onnxruntime/tools/ci_build/github/linux/docker/scripts/training/requirements.txt
edgchen1 6d5b93b805
Synchronize training dependency versions between Docker image and Python wheel. (#5261)
Synchronize training dependency versions between Docker image and wheel, update docs, refactor build scripts.
2020-09-23 19:03:42 -07:00

8 lines
222 B
Text

--pre
-f https://download.pytorch.org/whl/nightly/cu102/torch_nightly.html
# transformers requires sklearn
sklearn
transformers==v2.10.0
torch==1.6.0.dev20200610
torchvision==0.7.0.dev20200610
torchtext==0.6.0.dev20200610