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

12 lines
271 B
Text

--pre
-f https://download.pytorch.org/whl/nightly/cu101/torch_nightly.html
# transformers requires sklearn
pandas
sklearn
transformers==v2.10.0
torch==1.8.0.dev20201201+cu101
torchvision==0.9.0.dev20201201+cu101
torchtext==0.9.0.dev20201201
tensorboard==v2.0.0
h5py
wget