From 249a2c14effe546c0487d5793325e7fedaf68bc0 Mon Sep 17 00:00:00 2001 From: baijumeswani Date: Thu, 1 Apr 2021 09:37:47 -0700 Subject: [PATCH] Pin version of pytorch to 1.8.1 for ORTModule CI pipeline (#7167) * 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 --- .../training/ortmodule/stage1/requirements.txt | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements.txt b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements.txt index f9cef54640..1ce962f4ff 100644 --- a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements.txt +++ b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements.txt @@ -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 \ No newline at end of file +pytorch-lightning==1.2.5