Hello,

I noticed that in #3642 GPU support has been added, which is indeed much appreciated.

However, the current implementation always checks for GPUs availability and there's no way to turn this off. Users (like me) might need to do so, for example, when they are trying to run a pipeline on a CPU-only HPC (where no NVIDIA-SMI is installed).

Could we maybe have another parameter that would skip checking for GPUs? I would be up for making a PR if you think this makes sense. Please let me know.

Thanks!