From 4e9dd42663de1b9ff24b636f04185e2fe72dd777 Mon Sep 17 00:00:00 2001 From: chicm-ms Date: Thu, 19 Mar 2020 00:37:47 +0800 Subject: [PATCH] fix torch version --- test/pipelines/pipelines-it-local-windows.yml | 2 +- test/pipelines/pipelines-it-local.yml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/test/pipelines/pipelines-it-local-windows.yml b/test/pipelines/pipelines-it-local-windows.yml index a63165e719..7691ad19fe 100644 --- a/test/pipelines/pipelines-it-local-windows.yml +++ b/test/pipelines/pipelines-it-local-windows.yml @@ -9,7 +9,7 @@ jobs: - script: | python -m pip install scikit-learn==0.20.0 --user python -m pip install keras==2.1.6 --user - python -m pip install torch===1.13.1 torchvision===0.4.1 -f https://download.pytorch.org/whl/torch_stable.html --user + python -m pip install torchvision===0.4.1 torch===1.3.1 -f https://download.pytorch.org/whl/torch_stable.html --user python -m pip install tensorflow-gpu==1.11.0 --user displayName: 'Install dependencies for integration tests' - script: | diff --git a/test/pipelines/pipelines-it-local.yml b/test/pipelines/pipelines-it-local.yml index 7708285ec3..b0d385f630 100644 --- a/test/pipelines/pipelines-it-local.yml +++ b/test/pipelines/pipelines-it-local.yml @@ -10,8 +10,8 @@ jobs: displayName: 'Install nni toolkit via source code' - script: | python3 -m pip install scikit-learn==0.20.0 --user - python3 -m pip install torch==1.13.1 --user python3 -m pip install torchvision==0.4.1 --user + python3 -m pip install torch==1.3.1 --user python3 -m pip install keras==2.1.6 --user python3 -m pip install tensorflow-gpu==1.15 --user sudo apt-get install swig -y