Install nvidia toolkit for testing

This commit is contained in:
Cosmin Ciocan
2024-01-02 15:26:46 +01:00
parent 8ae5483629
commit f467060bcf
+6 -1
View File
@@ -27,7 +27,12 @@ jobs:
with:
python-version: ${{ matrix.python-version }}
- name: Install dependencies
- name: Install CUDA tools
run: |
apt update
apt install nvidia-cuda-toolkit
- name: Install Python dependencies
run: |
python -m pip install --upgrade pip
pip install -r tests/requirements.txt