From d5742f3a431bebc139318087ca126e1eb68fcaae Mon Sep 17 00:00:00 2001 From: Abhishek Jindal Date: Wed, 29 Dec 2021 19:35:10 -0800 Subject: [PATCH] moving from torch nightly build to stable build (#10150) * moving from torch nightly build to stable build * using torch cpu version * using torch cpu version from link --- .../ortmodule/stage1/requirements_torch_eager_cpu.txt | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_eager_cpu.txt b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_eager_cpu.txt index 8ffd732ece..487bfdbecf 100644 --- a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_eager_cpu.txt +++ b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_eager_cpu.txt @@ -1,10 +1,9 @@ --pre --f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html -torch>=1.9.0 +-f https://download.pytorch.org/whl/cpu/torch_stable.html +torch>=1.10.0 setuptools>=41.4.0 cerberus h5py sklearn numpy pandas -