3 ms·
Small code example in the PyTorch doc: https://pytorch.org/docs/master/notes/mps.html https://pytorch.org/docs/master/notes/mps.html
by MasterScrat 4y ago
Small code example in the PyTorch doc:
https://pytorch.org/docs/master/notes/mps.html https://pytorch.org/docs/master/notes/mps.html
- nxpnsv 4y agoTried https://pytorch.org/tutorials/beginner/basics/quickstart_tutorial.html https://pytorch.org/tutorials/beginner/basics/quickstart_tut... with mps vs cpu. mps worked, but cpu actually was faster (16 vs 21s). Perhaps I am doing it wrong...