Even if it was a good idea - how would you draw a boundary between AI and the rest of computing? In the end you'd have to go full Butlerian Jihad and prohibit computation in general, or at least limit allowed computing power.
This will of course be the policy set by physics once we have exhausted all the easily accessible energy reserves: limited everything power. Nature will conduct Butlerian jihad with no great zeal or ardor, but with just patience.
That's going to take a very long time, there's massive amounts of solar, geothermal, and wind energy left available. Hell, natural gas is a waste product of oil drilling.