  
      def _user_input(self) -> None:
          """ Get user input for AMD/DirectML/ROCm/Cuda/Docker """
          self._directml_ask_enable()
          if not self._env.backend:
              self._amd_ask_enable()
          if not self._env.backend:
              self._docker_ask_enable()
              self._cuda_ask_enable()
          if self._env.os_version[0] != "Linux" and (self._env.enable_docker
 
</s> remove         """ Get user input for AMD/Cuda/Docker """
 </s> add         """ Get user input for AMD/DirectML/ROCm/Cuda/Docker """ </s> remove     def update_tf_dep(self) -> None:
        """ Update Tensorflow Dependency """
        if self.is_conda or self.backend != "nvidia":
            # CPU/AMD doesn't need Cuda and Conda handles Cuda and cuDNN so nothing to do here
 </s> add     def _update_tf_dep_nvidia(self) -> None:
        """ Update the Tensorflow dependency for global Cuda installs """
        if self.is_conda:  # Conda handles Cuda and cuDNN so nothing to do here </s> remove         self._env.update_tf_dep()
 </s> add         self._check_rocm()
        self._rocm_ask_enable()
  
      def _user_input(self) -> None:
          """ Get user input for AMD/DirectML/ROCm/Cuda/Docker """
          self._directml_ask_enable()
        self._rocm_ask_enable()
          if not self._env.backend:
              self._amd_ask_enable()
          if not self._env.backend:
              self._docker_ask_enable()
              self._cuda_ask_enable()
          if self._env.os_version[0] != "Linux" and (self._env.enable_docker
keep keep keep add keep keep keep keep keep keep
 Linux - ROCm (AMD) support 
setup.py