May 22, 2022, 9:32 p.m. | /u/seraschka

Machine Learning www.reddit.com

If someone is curious, I updated the benchmarks after the PyTorch team fixed the memory leak in the latest nightly release May 21->22. The results are quite improved:

​

https://preview.redd.it/5dkat9hoi3191.png?width=2637&format=png&auto=webp&s=dc42ee03167dd3aefbd0319061994bfc2ff24dab

For a more detailed write-up please see [https://sebastianraschka.com/blog/2022/pytorch-m1-gpu.html](https://sebastianraschka.com/blog/2022/pytorch-m1-gpu.html)

benchmark gpu m1 machinelearning memory pytorch

Senior Machine Learning Engineer (MLOps)

@ Promaton | Remote, Europe

Lead Software Engineer - Artificial Intelligence, LLM

@ OpenText | Hyderabad, TG, IN

Lead Software Engineer- Python Data Engineer

@ JPMorgan Chase & Co. | GLASGOW, LANARKSHIRE, United Kingdom

Data Analyst (m/w/d)

@ Collaboration Betters The World | Berlin, Germany

Data Engineer, Quality Assurance

@ Informa Group Plc. | Boulder, CO, United States

Director, Data Science - Marketing

@ Dropbox | Remote - Canada