In the US, a driver license isn't a proof of citizenship. Also, state IDs are not accepted by federal agencies, so it probably wouldn't work as proof of citizenship on federal elections.
Federal elections are all run by the individual states, so a state ID would be all you need.
There really are not federal elections. We call them that, but they are state elections for federal office.