Most compatible Linux distro for NVIDIA GeForce 1050 TI and Tensorflow GPU

Hello,
I am moving to Linux for all my deep learning projects as I got a lot of great feedback about its stability.

I have GeForce 1050 Ti which is working well on Windows. I installed Linux Ubuntu 20.04 and the latest driver nvidia-driver-450.66 , TensorFlow, CUDA 10.1 and CUDNN 7.6.5. Everything is working well. The only problem is I see lag when scrolling browsing (like sluggish overlapping in the middle of screen during scrolling happens randomly).

I suspect it is a driver compatibility. THerefore, may I know the most compatible Linux with GeForce 1050 Ti that is known to work without problems while can operate on TensorFlow GPU?

Thanks