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 dea4665e..c34df22d 100644 --- a/tools/ci_build/github/linux/docker/scripts/training/requirements.txt +++ b/tools/ci_build/github/linux/docker/scripts/training/requirements.txt @@ -2,7 +2,7 @@ -f https://download.pytorch.org/whl/torch_stable.html # transformers requires sklearn sklearn -numpy==1.19.5 +numpy==1.22.0 transformers==v2.10.0 torch==1.6.0+cu101 torchvision==0.7.0+cu101