lab icon indicating copy to clipboard operation
lab copied to clipboard

Published 20 hours ago • verified publisher icon wesselb

Support for MPS (Mac GPU) when using torch backend.

Open magnusross opened this issue 1 year ago • 0 comments

Hey Wessel!

I am just wondering if it might be possible to add support for the MPS backend in PyTorch as an alternative to Cuda, which would allow for the use of lab on the GPU on Mac.

It is hopefully quite a simple change I think, I have tried an initial implementation #11

magnusross avatar Sep 01 '23 09:09 magnusross