Was there a ban on nuclear weapons after world war 2?

No. After WW2 nukes were not banned, they were more like mass produced by the US and the USSR, followed by UK and France.