Can I run an eGPU on a Linux Partition of a Macbook Pro?

I'm hoping to do some machine learning / tensorflow tasks with my eGPU, but after it arrived today, I realized that tensorflow discontinued support for Mac NVIDIA GPUs.

I'd like to install a Ubuntu partition and run the most recent tensorflow-gpu package on that. Would the eGPU still be supported in that case? Would any extra steps be needed to set it up in macOS before running it in Ubuntu?

If you install Ubuntu as a dual boot arrangement, both operating systems run independent of each other. In other words, when booted to Ubuntu, the Mac preforms the same as any other PC. The only involvement with the macOS would be the need to create enough free space to allow Ubuntu to be installed.