For machine learning, Nvidias driver situation is indeed terrible, but AMDs is worse in this sense: If you get the wrong kernel version, driver version or somehow mismatch the cuda version due to a pinned dependency, you're up the creek. However, everything downstream will work. With AMD, you will get the drivers on your machine no problem but heaven help you if you try to do anything with them (there are regressions and memory leaks, etc.)