(Incidentally, I’m currently doing all of my training on CPU. Here’s some additional info about the machine I’m currently running this on, if it helps:)
$ lscpu | grep -E "^(CPU\(s\):|Model name:)"
CPU(s):                48
Model name:            Intel(R) Xeon(R) Silver 4214 CPU @ 2.20GHz