Hacker Newsnew | past | comments | ask | show | jobs | submitlogin

Have you tried setting device="mps" to use Metal? It should be faster than PyTorch's "cpu" device on Mac.


Guidelines | FAQ | Lists | API | Security | Legal | Apply to YC | Contact

Search: