Re:16-inch MacBook Pro
by faizshah on Wednesday November 13, 2019 @09:35AM
(#21524508)
What is the state of deep learning on AMD? Is this GPU usable with pytorch?
[ Parent
| on HN ]
Pretty bad. There is pytorch for rocm (AMD's CUDA emulation) but I don't think anyone actually uses it.
https://github.com/ROCmSoftwarePlatform/pytorch/issues
I don't think you'll be doing much local ML work on this laptop
Yeah, generally a cloud based approach to training models is better than on your mac. AMD folks have been working on RoCM for some time and it works reasonably well for common models but the AMD GPU HW isn't pervasive like Nvidia GPUs in places like AWS.
If you are looking at using Colab for prototyping, you can also try TPUs which are now supported for PyTorch. Here is the link some additional info including some Colab NBs: https://github.com/pytorch/xla