| name: tests-cuda | |
| on: | |
| push: | |
| branches: | |
| - master | |
| pull_request: | |
| branches: | |
| - master | |
| schedule: | |
| - cron: "0 0 * * *" | |
| jobs: | |
| tests-cuda: | |
| container: | |
| image: pytorch/pytorch:1.6.0-cuda10.1-cudnn7-runtime | |
| options: --gpus all | |
| runs-on: [self-hosted, Linux, x64] | |
| steps: | |
| - uses: actions/checkout@v2 | |
| - name: Install dependencies | |
| shell: bash -l {0} | |
| run: | | |
| # NOTE: pytorch image does not include git | |
| conda install -c anaconda git | |
| pip install .[all] | |
| - name: Run Tests | |
| shell: bash -l {0} | |
| run: pytest -v --device cuda --dtype float32,float64 ./test | |