onnxruntime/tools/ci_build/github/linux/docker/scripts/training/requirements.txt
liqunfu 92662659ba
Liqun/remove number matching (#5606)
replace number matching with relaxed comparison in frontend tests
Co-authored-by: liqun <liqun@OrtTrainingDev4.af05slrtruoetgaxwwjv5nsq5e.px.internal.cloudapp.net>
2020-10-27 21:27:37 -07:00

10 lines
208 B
Text

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