Skip to content

Add CI pipeline for NVIDIA GPUs #7

Add CI pipeline for NVIDIA GPUs

Add CI pipeline for NVIDIA GPUs #7

name: Linux NVIDIA GPU tests
on:
push:
branches:
- main
paths:
- "quanto/**"
- "test/**"
- "pyproject.toml"
- "setup.py"
pull_request:
types: [assigned, opened, synchronize, reopened]
paths:
- "quanto/**"
- "test/**"
- "pyproject.toml"
- "setup.py"
jobs:
conventional-commits:
uses: ./.github/workflows/conventional-commits.yml
python-quality:
uses: ./.github/workflows/python-quality.yml
test-ubuntu:
needs: [conventional-commits, python-quality]
runs-on: [self-hosted, single-gpu , nvidia-gpu, a10, ci]
strategy:
fail-fast: false
matrix:
python-version: ["3.11"]
steps:
- uses: actions/checkout@v2
- uses: Jimver/[email protected]
id: cuda-toolkit
with:
cuda: '12.3.2'
- run: |
echo "Installed cuda version is: ${{steps.cuda-toolkit.outputs.cuda}}"
echo "Cuda install location: ${{steps.cuda-toolkit.outputs.CUDA_PATH}}"
nvcc -V
- name: Install CUDA Toolkit
shell: bash
run: |
wget --progress=bar:force:noscroll https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
sudo sh cuda_11.8.0_520.61.05_linux.run
nvcc --version
- name: Set up Python ${{ matrix.python-version }}
uses: actions/setup-python@v2
with:
python-version: ${{ matrix.python-version }}
# Install dependencies
- name: Configure and install dependencies
run: |
pip install --upgrade pip
pip install .[dev]
# Run tests
- name: Run tests
run: |
python -m pytest test