tch-rs icon indicating copy to clipboard operation
tch-rs copied to clipboard

Published 20 hours ago • verified publisher icon LaurentMazare

apple m1 GPU support

Open jianshu93 opened this issue 3 years ago • 0 comments

Dear torch rust team,

The newest version of PyTorch (https://pytorch.org/blog/introducing-accelerated-pytorch-training-on-mac/) also support apple M1 devices via the MPS interface provided by apple. I am wondering whether this is also doable here. Libtorch version 1.12 can be installed and compiled on M1 device but I do not know how the GPU was supported (via universal binary or something related to that).

Thanks,

Jianshu

jianshu93 avatar Aug 09 '22 22:08 jianshu93