From e48b9803bedeb2f83862f8b9ee4697ec9389d1d0 Mon Sep 17 00:00:00 2001 From: Wil Brady <25513670+WilBrady@users.noreply.github.com> Date: Wed, 29 Jun 2022 15:51:46 -0400 Subject: [PATCH] Fix windows and linux eager build to torch 1.11.0. --- .../training/ortmodule/stage1/torch_eager_cpu/requirements.txt | 2 +- tools/ci_build/github/windows/eager/requirements.txt | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/torch_eager_cpu/requirements.txt b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/torch_eager_cpu/requirements.txt index 487bfdbecf214..5858a7b27362a 100644 --- a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/torch_eager_cpu/requirements.txt +++ b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/torch_eager_cpu/requirements.txt @@ -1,6 +1,6 @@ --pre -f https://download.pytorch.org/whl/cpu/torch_stable.html -torch>=1.10.0 +torch==1.11.0 setuptools>=41.4.0 cerberus h5py diff --git a/tools/ci_build/github/windows/eager/requirements.txt b/tools/ci_build/github/windows/eager/requirements.txt index 557d5efd0b61d..508e0c0d78ab3 100644 --- a/tools/ci_build/github/windows/eager/requirements.txt +++ b/tools/ci_build/github/windows/eager/requirements.txt @@ -2,5 +2,4 @@ setuptools wheel numpy typing_extensions -torch - +torch==1.11.0