Do you have the same problem with:
  
      

      TensorFlow
  

  
    

tf.config.experimental.set_memory_growth  |  TensorFlow Core v2.6.0 7

  Set if memory growth should be enabled for a PhysicalDevice.