AttributeError: module 'tensorflow' has no attribute 'function' about retinaface HOT 3 CLOSED

Looking in indexes: https://pypi.douban.com/simple
ERROR: Could not find a version that satisfies the requirement tensorflow==2.7.0
ERROR: No matching distribution found for tensorflow==2.7.0

---------------------------------------------------------------------------
InternalError                             Traceback (most recent call last)
<ipython-input-1-ee0c71cae24a> in <module>
      5 os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'
      6 
----> 7 faces = RetinaFace.extract_faces("ai.jpg")
      8 
      9 print("There are ", len(faces), " faces in the image")

/usr/local/lib/python3.6/site-packages/retinaface/RetinaFace.py in extract_faces(img_path, threshold, model, align, allow_upscaling)
    192     #---------------------------
    193 
--> 194     obj = detect_faces(img_path = img, threshold = threshold, model = model, allow_upscaling = allow_upscaling)
    195 
    196     if type(obj) == dict:

/usr/local/lib/python3.6/site-packages/retinaface/RetinaFace.py in detect_faces(img_path, threshold, model, allow_upscaling)
     88     landmarks_list = []
     89     im_tensor, im_info, im_scale = preprocess.preprocess_image(img, allow_upscaling)
---> 90     net_out = model(im_tensor)
     91     net_out = [elt.numpy() for elt in net_out]
     92     sym_idx = 0

/usr/local/lib/python3.6/site-packages/tensorflow/python/eager/def_function.py in __call__(self, *args, **kwds)
    883 
    884       with OptionalXlaContext(self._jit_compile):
--> 885         result = self._call(*args, **kwds)
    886 
    887       new_tracing_count = self.experimental_get_tracing_count()

/usr/local/lib/python3.6/site-packages/tensorflow/python/eager/def_function.py in _call(self, *args, **kwds)
    955       # If we did not create any variables the trace we have is good enough.
    956       return self._concrete_stateful_fn._call_flat(
--> 957           filtered_flat_args, self._concrete_stateful_fn.captured_inputs)  # pylint: disable=protected-access
    958 
    959     def fn_with_cond(inner_args, inner_kwds, inner_filtered_flat_args):

/usr/local/lib/python3.6/site-packages/tensorflow/python/eager/function.py in _call_flat(self, args, captured_inputs, cancellation_manager)
   1962       # No tape is watching; skip to running the function.
   1963       return self._build_call_outputs(self._inference_function.call(
-> 1964           ctx, args, cancellation_manager=cancellation_manager))
   1965     forward_backward = self._select_forward_and_backward_functions(
   1966         args,

/usr/local/lib/python3.6/site-packages/tensorflow/python/eager/function.py in call(self, ctx, args, cancellation_manager)
    594               inputs=args,
    595               attrs=attrs,
--> 596               ctx=ctx)
    597         else:
    598           outputs = execute.execute_with_cancellation(

/usr/local/lib/python3.6/site-packages/tensorflow/python/eager/execute.py in quick_execute(op_name, num_outputs, inputs, attrs, ctx, name)
     58     ctx.ensure_initialized()
     59     tensors = pywrap_tfe.TFE_Py_Execute(ctx._handle, device_name, op_name,
---> 60                                         inputs, attrs, num_outputs)
     61   except core._NotOkStatusException as e:
     62     if name is not None:

InternalError:  cuDNN launch failure : input shape ([1,3,1019,1980])
	 [[node model/bn_data/FusedBatchNormV3 (defined at /usr/local/lib/python3.6/site-packages/retinaface/RetinaFace.py:90) ]] [Op:__inference_function_7611]

Function call stack:
function