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.
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
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