Get a penis pump to allow you to get erections before atrophy. It will allow you get fully erect it just won’t be as stiff as you need for sex.

Were they able to spare the nerves when they did the prostatectomy? If not you may need an implant or the bimix or Trimix shots.