search

ROCm / tensorflow-upstream

TensorFlow ROCm port
https://tensorflow.org
Apache License 2.0
686 stars 94 forks source link

Does tensorflow ROCm support Tensorflow Serving? #1869

Open ndeepesh opened 2 years ago

ndeepesh commented 2 years ago

Issue Type

Others

Source

source

Tensorflow Version

tf 2.4

Custom Code

No

OS Platform and Distribution

Linux

Mobile device

No response

Python version

3.8

Bazel version

No response

GCC/Compiler version

No response

CUDA/cuDNN version

No response

GPU model and memory

No response

Current Behaviour?

Question - Is TF Serving supported for AMD GPU's aka ROCm?

Standalone code to reproduce the issue

NA

Relevant log output

No response

jayfurmanek commented 2 years ago

It does work, but currently you have to build it yourself. I've created a dockerfile that will do that if you are interested:

https://github.com/jayfurmanek/serving/tree/rocm53