From fdf12a5c3516a7a28fb2f40a446a65b5e7f0acc5 Mon Sep 17 00:00:00 2001 From: Wil Brady <25513670+WilBrady@users.noreply.github.com> Date: Thu, 30 Jun 2022 07:01:13 -0400 Subject: [PATCH] Fix windows eager build break by pinning to torch version 1.11.0 (#12033) 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 487bfdbecf..5858a7b273 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 557d5efd0b..508e0c0d78 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