For AMD GPU check:

  
    
    
    Use AMD GPU for model training General Discussion
  
  
    Hello folks, 
I have got an AMD GPU on my system and I want to use it for model training. After several efforts I have not been able to do so. 
Since CUDA is only supported on NVIDIA GPU, it doesn’t work for me. 
Any suggestions? 
Thank you.