3 ms·
No cpu / mps support to train on Macs, apparently.
by antirez 2y ago
No cpu / mps support to train on Macs, apparently.
- leopoldj 2y agoYou should be able to make a few small changes to support "mps". In TrainingConfig set the device to "mps". The run training. In sample.py modify parse_args() and add support for mps as a possible value for the --device argument.
- antirez 2y agoThanks! I'll try. I didn't bother believing that if this was developed heavily on CUDA, it was likely going to use kernels that were missing in MPS.