Citation: 9C_623/2020 E. C

A.________ lässt mit Beschwerde in öffentlich-rechtlichen Angelegenheiten beantragen, unter Aufhebung des Entscheids vom 18. August 2020 und der Verfügung vom 31. März 2020 sei ihm Umschulung zu gewähren und die Sache sei zur Prüfung und Festlegung geeigneter Mass nahmen und der weiteren im Zusammenhang stehenden gesetzlichen Leistungen an die IV-Stelle zurückzuweisen. Eventualiter sei die Sache zur weiteren medizinischen Abklärung und neuem Entscheid an die IV-Stelle, subeventualiter an das kantonale Gericht, zurückzuweisen. Die IV-Stelle schliesst auf Abweisung der Beschwerde. Das Bundesamt für Sozialversicherungen verzichtet auf eine Vernehmlassung.