62 lines
1.5 KiB
YAML
62 lines
1.5 KiB
YAML
name: Self-hosted runner (push)
|
|
|
|
on:
|
|
push:
|
|
branches:
|
|
- master
|
|
paths:
|
|
- "src/**"
|
|
- "tests/**"
|
|
- ".github/**"
|
|
# pull_request:
|
|
repository_dispatch:
|
|
|
|
|
|
jobs:
|
|
run_tests_torch_and_tf_gpu:
|
|
runs-on: self-hosted
|
|
steps:
|
|
- uses: actions/checkout@v2
|
|
- name: Python version
|
|
run: |
|
|
which python
|
|
python --version
|
|
pip --version
|
|
- name: Current dir
|
|
run: pwd
|
|
- run: nvidia-smi
|
|
- name: Create new python env (on self-hosted runners we have to handle isolation ourselves)
|
|
run: |
|
|
python -m venv .env
|
|
source .env/bin/activate
|
|
which python
|
|
python --version
|
|
pip --version
|
|
- name: Install dependencies
|
|
run: |
|
|
source .env/bin/activate
|
|
pip install torch
|
|
pip install .[sklearn,testing]
|
|
|
|
- name: Are GPUs recognized by our DL frameworks
|
|
run: |
|
|
source .env/bin/activate
|
|
python -c "import torch; print(torch.cuda.is_available())"
|
|
|
|
- name: Run all non-slow tests on GPU
|
|
env:
|
|
TF_FORCE_GPU_ALLOW_GROWTH: "true"
|
|
# TF_GPU_MEMORY_LIMIT: 4096
|
|
OMP_NUM_THREADS: 1
|
|
USE_CUDA: yes
|
|
run: |
|
|
source .env/bin/activate
|
|
python -m pytest -n 2 --dist=loadfile -s ./tests/ | tee output.txt
|
|
- name: cat output.txt
|
|
run: cat output.txt
|
|
- name: Upload output.txt
|
|
uses: actions/upload-artifact@v1
|
|
with:
|
|
name: pytest_output
|
|
path: output.txt
|