Run tests on GPU if available

Currently integration tests run on CPU (https://gitlab.cern.ch/atlas-flavor-tagging-tools/algorithms/salt/-/blob/main/salt/tests/test_pipeline.py?ref_type=heads#L38).

We might be able to speed up the test suite a bit when running locally by using a GPU if one is available.