Skip to content

Code Quality Improvements and Development Infrastructure Updates #6

Code Quality Improvements and Development Infrastructure Updates

Code Quality Improvements and Development Infrastructure Updates #6

Workflow file for this run

name: GPU Tests
on:
push:
branches: [ master ]
pull_request:
branches: [ master ]
jobs:
gpu-test:
runs-on: ubuntu-20.04
steps:
- uses: actions/checkout@v3
- name: Set up Python 3.7
uses: actions/setup-python@v4
with:
python-version: '3.7'
- name: Install FFMPEG
run: |
sudo apt-get update
sudo apt-get install -y ffmpeg
- name: Install PySide2
run: |
python -m pip install --upgrade pip
pip install "pyside2==5.13.2"
- name: Install PyTorch with CUDA
run: |
pip install torch==1.11.0+cu115 torchvision==0.12.0+cu115 -f https://download.pytorch.org/whl/torch_stable.html
- name: Install package and test dependencies
run: |
python -m pip install --upgrade "pip<24.0"
pip install -r requirements.txt
pip install pytest pytest-cov
python setup.py develop
- name: GPU Tests
run: |
pytest -v -m "gpu" tests/
env:
CUDA_VISIBLE_DEVICES: 0