saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-20 19:12:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Pin version of pytorch to 1.8.1 for ORTModule CI pipeline (#7167)

Browse source 
* Pin version of pytorch to 1.8.1 for ORTModule CI pipeline
* Use pytorch-lightning stable version 1.2.5
* Revert to cuda 10.1
This commit is contained in:
baijumeswani 2021-04-01 09:37:47 -07:00 committed by GitHub
parent fc6ac5bfac
commit 249a2c14ef
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements.txt
View file

@ -1,15 +1,14 @@
--pre
-f https://download.pytorch.org/whl/nightly/cu101/torch_nightly.html
-f https://download.pytorch.org/whl/torch_stable.html
# transformers requires sklearn
pandas
sklearn
numpy==1.19.5
transformers==v4.3.2
torch
torchvision
torchtext
torch==1.8.1+cu101
torchvision==0.9.1+cu101
torchtext==0.9.1
tensorboard==v2.0.0
h5py
wget
# PyTorch Lightning (nightly) is used for CI tests only
https://github.com/PyTorchLightning/pytorch-lightning/archive/master.zip
pytorch-lightning==1.2.5