We need to have a custom logic that uses the new layout if available, and fall back to the old layout. Something like this:
try:
from cuda.bindings import driver, runtime, nvrtc
except ImportError:
from cuda import cuda as driver
from cuda import cudart as runtime
from cuda import nvrtc
It's the best to centralize this logic in, say, _utils.py and have all other modules import from there.