From e35db194e3e8fad2a73e5f856fd68b75c574d45c Mon Sep 17 00:00:00 2001 From: Guoyu Wang <62914304+gwang-msft@users.noreply.github.com> Date: Thu, 14 Jan 2021 00:33:22 -0800 Subject: [PATCH] fix the pipeline failure (#6346) --- .../github/linux/docker/scripts/training/requirements.txt | 1 + 1 file changed, 1 insertion(+) diff --git a/tools/ci_build/github/linux/docker/scripts/training/requirements.txt b/tools/ci_build/github/linux/docker/scripts/training/requirements.txt index 5668c537cd..0af37059a3 100644 --- a/tools/ci_build/github/linux/docker/scripts/training/requirements.txt +++ b/tools/ci_build/github/linux/docker/scripts/training/requirements.txt @@ -2,6 +2,7 @@ -f https://download.pytorch.org/whl/torch_stable.html # transformers requires sklearn sklearn +numpy==1.16.6 transformers==v2.10.0 torch==1.6.0+cu101 torchvision==0.7.0+cu101