remove torch (#2172)

microsoft · Mar 18, 2020 · e39958f · e39958f
1 parent 0c04e78
commit e39958f
Show file tree

Hide file tree

Showing 7 changed files with 0 additions and 12 deletions.
diff --git a/test/nni_test/nnitest/run_tests.py b/test/nni_test/nnitest/run_tests.py
@@ -9,7 +9,6 @@
 import shlex
 import traceback
 import json
-import torch
 import ruamel.yaml as yaml
 
 from utils import setup_experiment, get_experiment_status, get_yml_content, dump_yml_content, get_experiment_id, \
@@ -59,11 +58,6 @@ def run_test_case(test_case_config, it_config, args):
     if sys.platform == 'win32':
         test_yml_config['trial']['command'] = test_yml_config['trial']['command'].replace('python3', 'python')
 
-    # check GPU
-    if test_yml_config['trial']['gpuNum'] > 0 and torch.cuda.device_count() < 1:
-        print('skipping {}, gpu is not available'.format(test_case_config['name']))
-        return
-
     # generate temporary config yml file to launch experiment
     new_config_file = config_path + '.tmp'
     dump_yml_content(new_config_file, test_yml_config)

diff --git a/test/pipelines/pipelines-it-frameworkcontroller.yml b/test/pipelines/pipelines-it-frameworkcontroller.yml
@@ -22,7 +22,6 @@ jobs:
     displayName: 'Install nni toolkit via source code'
 
   - script: |
-      python3 -m pip install torch==1.2.0 --user
       sudo apt-get install swig -y
       PATH=$HOME/.local/bin:$PATH nnictl package install --name=SMAC
       PATH=$HOME/.local/bin:$PATH nnictl package install --name=BOHB

diff --git a/test/pipelines/pipelines-it-kubeflow.yml b/test/pipelines/pipelines-it-kubeflow.yml
@@ -22,7 +22,6 @@ jobs:
     displayName: 'Install nni toolkit via source code'
 
   - script: |
-      python3 -m pip install torch==1.2.0 --user
       sudo apt-get install swig -y
       PATH=$HOME/.local/bin:$PATH nnictl package install --name=SMAC
       PATH=$HOME/.local/bin:$PATH nnictl package install --name=BOHB

diff --git a/test/pipelines/pipelines-it-pai-windows.yml b/test/pipelines/pipelines-it-pai-windows.yml
@@ -58,7 +58,6 @@ jobs:
   - script: |
       set PATH=$(ENV_PATH)
       python -m pip install scikit-learn==0.21.0 --user
-      python -m pip install torch===1.2.0 -f https://download.pytorch.org/whl/torch_stable.html --user
     displayName: 'Install dependencies for integration tests'
   - script: |
       cd test

diff --git a/test/pipelines/pipelines-it-pai.yml b/test/pipelines/pipelines-it-pai.yml
@@ -22,7 +22,6 @@ jobs:
     displayName: 'Install nni toolkit via source code'
 
   - script: |
-      python3 -m pip install torch==1.2.0 --user
       sudo apt-get install swig -y
       PATH=$HOME/.local/bin:$PATH nnictl package install --name=SMAC
       PATH=$HOME/.local/bin:$PATH nnictl package install --name=BOHB

diff --git a/test/pipelines/pipelines-it-remote-windows.yml b/test/pipelines/pipelines-it-remote-windows.yml
@@ -17,7 +17,6 @@ jobs:
     displayName: 'Install nni toolkit via source code'
   - script: |
       python -m pip install scikit-learn==0.20.1 --user
-      python -m pip install torch===1.2.0 -f https://download.pytorch.org/whl/torch_stable.html --user
     displayName: 'Install dependencies for integration tests'
   - task: SSH@0
     inputs:

diff --git a/test/pipelines/pipelines-it-remote.yml b/test/pipelines/pipelines-it-remote.yml
@@ -15,7 +15,6 @@ jobs:
       source install.sh
     displayName: 'Install nni toolkit via source code'
   - script: |
-      python3 -m pip install torch==1.2.0 --user
       sudo apt-get install swig -y
       PATH=$HOME/.local/bin:$PATH nnictl package install --name=SMAC
       PATH=$HOME/.local/bin:$PATH nnictl package install --name=BOHB