Can you check if the GPU is visibile?

  
      

      TensorFlow
  

  
    

tf.config.list_physical_devices  |  TensorFlow Core v2.6.0 10

  Return a list of physical devices visible to the host runtime.


  

  
    
    
  

  



  
      

      TensorFlow
  

  
    

tf.config.list_logical_devices  |  TensorFlow Core v2.6.0 5

  Return a list of logical devices created by runtime.