xtalker icon indicating copy to clipboard operation
xtalker copied to clipboard

Published 20 hours ago • verified publisher icon Spycsh

MPS device compatibility

Open adills opened this issue 1 year ago • 2 comments

With Pytorch and TF both supporting Apple's M1/M2 silicon chips, you should be able to include a device checker and include "mps" capability for GPU processing. If only I had the time to create a branch and show you.

adills avatar Sep 15 '23 11:09 adills