Does tensorflow support AMD GPUs through ROCm?

TensorFlow General Discussion
1

There is no mention in the installation guides but I noticed there is a tensorflow-rocm package, yet it uses 2.14 and may not be a good idea to try it.

3

Hi @Mah_Neh, As tensorflow utilizes nvidia toolkit for utilizing GPU, this toolkit was specifically designed for Nvidia gpu’s. I think this tf-rocm was developed and maintained by AMD for using Tensorflow with AMD GPU, so this might be the reason for not mentioning it in the official document. Thank You.

1 Like
5

thanks for your reply, it seems the port is here GitHub - ROCm/tensorflow-upstream: TensorFlow ROCm port

still no support for 2.15, but interesting that there is smth.

just realised i can’t mark yours as an answer ? @Kiran_Sai_Ramineni