File tree Expand file tree Collapse file tree 1 file changed +36
-0
lines changed Expand file tree Collapse file tree 1 file changed +36
-0
lines changed Original file line number Diff line number Diff line change
1
+ name : Run Horovod tests
2
+ on : [push, pull_request]
3
+
4
+ jobs :
5
+ horovod-tests :
6
+ runs-on : ubuntu-latest
7
+ strategy :
8
+ matrix :
9
+ python-version : [3.7]
10
+ pytorch-channel : [pytorch]
11
+
12
+
13
+ steps :
14
+ - uses : actions/checkout@v2
15
+ - name : Setup Miniconda
16
+ uses : goanpeca/setup-miniconda@v1
17
+ with :
18
+ miniconda-version : " latest"
19
+ python-version : ${{ matrix.python-version }}
20
+
21
+ - name : Install dependencies
22
+ shell : bash -l {0}
23
+ run : |
24
+ #install other dependencies
25
+ conda install pytorch torchvision cpuonly -c ${{ matrix.pytorch-channel }}
26
+ pip install -r requirements-dev.txt
27
+ # Fixes #1153
28
+ pip install --upgrade scipy==1.4.1
29
+ pip install horovod
30
+ python setup.py install
31
+
32
+ - name : Run Tests
33
+ shell : bash -l {0}
34
+ run : |
35
+ SKIP_DISTRIB_TESTS=${{ matrix.skip-distrib-tests }} CI_PYTHON_VERSION="${{ matrix.python-version }}" sh tests/run_cpu_tests.sh
36
+
You can’t perform that action at this time.
0 commit comments