The AI boom has sparked a race for chips, electricity and talent. Another potential constraint has appeared: water.