AI's water use is a problem, but shifting from electricity to solar or wind power could help

It depends. Though water consumption by data centers pales in comparison to that of some other industries, it's a problem in water-stressed regions. Water-saving cooling technologies could help.