Etter krigen rykket Ole Berg raskt opp i gradene og ble som generalløytnant 1. september 1945 sjef for Hærens overkommando og senere sjef for Forsvarsstaben (i praksis Forsvarssjef) i perioden 1. august 1946 – 31. oktober 1955.