I’m on an Ubuntu derivative called Mint, and on the first boot it gave me a pop up from the driver tool recommending that I change to the proprietary driver with an option for one click automatic download and install.
You are correct that this is detected and handled.