Accelerated PyTorch Training on M1 Mac

Re:Accelerated PyTorch Training on M1 Mac by arecurrence on Wednesday May 18, 2022 @11:56AM (#31424326)

This is much nicer ergonomics than what I had to do for tensorflow. It’s ostensibly out of the box support as a different torch device.

[ Parent | on HN ]

Re:Accelerated PyTorch Training on M1 Mac by mark_l_watson on Wednesday May 18, 2022 @11:59AM (#31424364)

I agree. I appreciated the M1/Metal TensorFlow support, but that was not as easy to setup.

[ Parent | on HN ]

Re:Accelerated PyTorch Training on M1 Mac by alfalfasprout on Wednesday May 18, 2022 @03:23PM (#31426609)

I mean, building tensorflow is generally an awful experience.

[ Parent | on HN ]

Re:Accelerated PyTorch Training on M1 Mac by dangrie158 on Wednesday May 18, 2022 @11:59AM (#31424365)

You must have installed it a while ago then :). I just recently did and only needed to I stall 2 packages via pip (i think tensoflow-macos and tenderfoot-metal ) which I found much better than wrangling with cuda and cudnn versions for Nvidia cards

[ Parent | on HN ]