I'd like to do that too, but currently you have to pick between games and AI stuff, at least on Linux. I'd love to support AMD on this, but their CUDA story isn't great.
True, but I considered a bit my needs and concluded that smooth Linux support is much more important for me than AI. My work doesn't directly touch AI, so for my modest AI needs I can just use online services that probably do it better anyway.