Retrieve The Oldest and Newest Supported CUDA Versions

Feature request

The STUMPY Python package depends on numba and, when cuda is available, it imports submodules that rely on cuda. It checks this by doing something like:

from numba import cuda

if cuda.is_available():  # Only import if CUDA is available
    from stumpy import gpu_stump

However, numba does not currently support CUDA 11.2 and if CUDA 11.2 is installed then the above if condition still returns True. It would be nice to have function that checks if the available CUDA version is compatible with numba like:

if cuda.is_available() and cuda.version_is_supported():
    from stumpy import gpu_stump

Or, as discussed with @stuartarchibald, maybe it makes more sense to add max/min supported versions enumerated and then make that part of the cuda.is_available() check.

[seibert]

I would also agree with the approach of combining the version check with the driver check in cuda.is_available() to keep things simple for downstream libraries.

[seanlaw]

Fantastic! Thanks, @gmarkall - please feel free to close this issue