Unverified Commit 9790e979 authored by chicm-ms's avatar chicm-ms Committed by GitHub
Browse files

Fix integration test dependencies (#822)

* Install dependencies for PAI/k8s IT

* updates
parent 7108466c
...@@ -38,6 +38,16 @@ jobs: ...@@ -38,6 +38,16 @@ jobs:
source install.sh source install.sh
displayName: 'Install nni toolkit via source code' displayName: 'Install nni toolkit via source code'
- script: |
python3 -m pip install scikit-learn==0.20.0 --user
python3 -m pip install torch==0.4.1 --user
python3 -m pip install torchvision==0.2.1 --user
python3 -m pip install keras==2.1.6 --user
python3 -m pip install tensorflow-gpu==1.10.0 --user
sudo apt-get install swig -y
nnictl package install --name=SMAC
displayName: 'Install dependencies for integration tests in Kubeflow mode'
- script: | - script: |
if [ $(build_docker_img) = 'true' ] if [ $(build_docker_img) = 'true' ]
then then
......
...@@ -38,6 +38,16 @@ jobs: ...@@ -38,6 +38,16 @@ jobs:
source install.sh source install.sh
displayName: 'Install nni toolkit via source code' displayName: 'Install nni toolkit via source code'
- script: |
python3 -m pip install scikit-learn==0.20.0 --user
python3 -m pip install torch==0.4.1 --user
python3 -m pip install torchvision==0.2.1 --user
python3 -m pip install keras==2.1.6 --user
python3 -m pip install tensorflow-gpu==1.10.0 --user
sudo apt-get install swig -y
nnictl package install --name=SMAC
displayName: 'Install dependencies for integration tests in PAI mode'
- script: | - script: |
if [ $(build_docker_img) = 'true' ] if [ $(build_docker_img) = 'true' ]
then then
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment