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 -