@Caspar Cui, Thanks for posting in in Q&A. Your thought is OK. You can manually create a registry key after the driver installation. Here is a link with an example for your reference.
https://hudson.tel/post/mass-driver-installation-using-powershell-and-intune/
Note: non-Microsoft link, just for the reference.
Hope the above information can help.
If the answer is helpful, please click "Accept Answer" and kindly upvote it. If you have extra questions about this answer, please click "Comment".
Note: Please follow the steps in our documentation to enable e-mail notifications if you want to receive the related email notification for this thread.