1infiniteloop t1_iunkkee wrote on November 1, 2022 at 6:04 PM Reply to [D] Machine learning prototyping on Apple silicon? by laprika0 Tensorflow models seem to work well on GPU (M1 Pro 32 GB memory). With Pytorch I have had difficulty getting the model to train on the GPU. CPU training for Pytorch models has been very slow. Permalink 1
1infiniteloop t1_iunkkee wrote
Reply to [D] Machine learning prototyping on Apple silicon? by laprika0
Tensorflow models seem to work well on GPU (M1 Pro 32 GB memory). With Pytorch I have had difficulty getting the model to train on the GPU. CPU training for Pytorch models has been very slow.