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