try change torch version
This commit is contained in:
Родитель
57bb1a5e9a
Коммит
e904d1aaa0
|
@ -29,15 +29,15 @@ jobs:
|
|||
~/.nn_meter
|
||||
/home/runner/work/nn-Meter/data/testmodels
|
||||
/opt/hostedtoolcache/Python/3.6.10/x64/lib/python3.6/site-packages
|
||||
key: ${{hashFiles('nn_meter/configs/predictors.yaml')}}-${{hashFiles('tests/integration_test.py')}}
|
||||
key: Data-${{hashFiles('nn_meter/configs/predictors.yaml')}}-${{hashFiles('tests/integration_test.py')}}
|
||||
|
||||
- name: Install dependencies
|
||||
if: steps.cache.outputs.cache-hit != 'true'
|
||||
run: |
|
||||
pip install tensorflow==1.15.0
|
||||
pip install onnx==1.9.0
|
||||
pip install torch==1.7.1
|
||||
pip install torchvision==0.8.2
|
||||
pip install torch==1.9.0
|
||||
pip install torchvision==0.10.0
|
||||
pip install onnx-simplifier
|
||||
|
||||
- name: Install nn-Meter
|
||||
|
|
|
@ -29,15 +29,15 @@ jobs:
|
|||
~/.nn_meter
|
||||
/home/runner/work/nn-Meter/data/testmodels
|
||||
/opt/hostedtoolcache/Python/3.6.10/x64/lib/python3.6/site-packages
|
||||
key: ${{hashFiles('nn_meter/configs/predictors.yaml')}}-${{hashFiles('tests/integration_test.py')}}
|
||||
key: Data-${{hashFiles('nn_meter/configs/predictors.yaml')}}-${{hashFiles('tests/integration_test.py')}}
|
||||
|
||||
- name: Install dependencies
|
||||
if: steps.cache.outputs.cache-hit != 'true'
|
||||
run: |
|
||||
pip install tensorflow==1.15.0
|
||||
pip install onnx==1.9.0
|
||||
pip install torch==1.7.1
|
||||
pip install torchvision==0.8.2
|
||||
pip install torch==1.9.0
|
||||
pip install torchvision==0.10.0
|
||||
pip install onnx-simplifier
|
||||
|
||||
- name: Install nni
|
||||
|
|
|
@ -29,15 +29,15 @@ jobs:
|
|||
~/.nn_meter
|
||||
/home/runner/work/nn-Meter/data/testmodels
|
||||
/opt/hostedtoolcache/Python/3.6.10/x64/lib/python3.6/site-packages
|
||||
key: ${{hashFiles('nn_meter/configs/predictors.yaml')}}-${{hashFiles('tests/integration_test.py')}}
|
||||
key: Data-${{hashFiles('nn_meter/configs/predictors.yaml')}}-${{hashFiles('tests/integration_test.py')}}
|
||||
|
||||
- name: Install dependencies
|
||||
if: steps.cache.outputs.cache-hit != 'true'
|
||||
run: |
|
||||
pip install tensorflow==1.15.0
|
||||
pip install onnx==1.9.0
|
||||
pip install torch==1.7.1
|
||||
pip install torchvision==0.8.2
|
||||
pip install torch==1.9.0
|
||||
pip install torchvision==0.10.0
|
||||
pip install onnx-simplifier
|
||||
|
||||
- name: Install nn-Meter
|
||||
|
|
Загрузка…
Ссылка в новой задаче