Can I train a Convolutional Neural Network optimized to AMD GPU

Can I train an optimized CNN on AMD GPU?

Hey John,

Most DL frameworks (including PyTorch and fastai) rely on CUDA, which only works with NVIDIA GPUs. There is some work in progress to improve the state of GPU acceleration for AMD, but currently it’s nowhere near as good as CUDA - so the bottom line is no, you cannot do so with fastai and I don’t think AMD GPUs will be competitive with NVIDIA for deep learning any time soon.

2 Likes

Q: Can I train an optimized CNN on AMD GPU?:
A: Yes, but not with fastai. (there is a pytorch compiled with ROC support somewhere)

1 Like

How ? Thanks