You need to use TF AMD ROCm

  
      

      github.com
  

  
    

ROCmSoftwarePlatform/tensorflow-upstream - Tensorflow ROCm port 241


  TensorFlow ROCm port. Contribute to ROCmSoftwarePlatform/tensorflow-upstream development by creating an account on GitHub.