AnsweredAssumed Answered

Tensorflow with AMD GPU

Question asked by visheshchanana on Jun 17, 2019
Latest reply on Jun 18, 2019 by dipak

I just bought a new Desktop with Ryzen 5 CPU and an AMD GPU to learn GPU programming. I am also interested in learning Tensorflow for deep neural networks. 

After a few days of fiddling with tensorflow on CPU, I realized I should shift all the computations to GPU. The tensorflow-gpu library isn't built for AMD as it uses CUDA while the openCL library cannot be used with tensorflow(I guess). 

 

Is there a library that I could use for tensorflow computations on AMD GPU??

Outcomes