There are many such schools, such as Dalian Railway Driver School, Guangzhou Railway Vocational and Technical College, Hunan Railway Vocational and Technical College and Guiyang Railway Driver School, whose major is electric locomotive crew.
The related major is locomotive major.
It should be noted that myopia doesn't matter, and color blindness certainly won't do. Besides, the work of train drivers is monotonous and hard.
However, you can't open it immediately after learning it!
Even if you can enter the locomotive depot after graduation, you usually have to study as a trainee driver (assistant driver) for at least half a year and three years before you can test the driver, and you have to receive at least two months of orientation education for new employees before you can take the car with you. Besides, you can't get into the locomotive depot if you study motorcycle now, unless you catch up with them or your relationship is hard enough.