Related and interesting thing - China appears to be building a trial reactor in the Gobi desert [0] because they think they have air cooling sorted out.
Or at least close enough. I haven't checked if it is technically in the desert.
There is still plenty of water, the problem is that you have to limit heating the river. With a cooling tower can cool with or without evaporating water depending on the design but always without heating the river. But it is cheaper to directly dump the heat into the water.
Dry cooling tower exists but they are more expensive. Natural cooling with the evaporation eventually runs into the problem due to built-up of salts which forces you to replace non-evaporated cooling water with fresh river water and lets you dump the heated water into the river (which is limiting factor due to environmental concerns). The less water is flowing and the higher the temperature of the river are the sooner this point is reached
My hunch is evaporative cooling uses not very much water. You could also fill a small dam when the water flow is higher. All means higher cost, of course.
There are detailed diagrams and formulas to calculate that. Engineering, not back of the envelope highschool physics.
But guess what, people designing an building power plants know this. And whatever is built is the best compromise possible at the time. Backnof envelope calculus in 2023 or not.