Graph execution error #74

Traceback (most recent call last):
File "/content/drive/.shortcut-targets-by-id/1ZcQQ8EgI0VKLZ8yrxSlsha1g3DZpxaKU/dr_pourreza/HRN/demo.py", line 78, in
run_hrn(args)
File "/content/drive/.shortcut-targets-by-id/1ZcQQ8EgI0VKLZ8yrxSlsha1g3DZpxaKU/dr_pourreza/HRN/demo.py", line 26, in run_hrn
output = reconstructor.predict(img, visualize=True, save_name=save_name, out_dir=out_dir)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/content/drive/.shortcut-targets-by-id/1ZcQQ8EgI0VKLZ8yrxSlsha1g3DZpxaKU/dr_pourreza/HRN/models/hrn.py", line 238, in predict
output = self.predict_base(img)
^^^^^^^^^^^^^^^^^^^^^^
File "/content/drive/.shortcut-targets-by-id/1ZcQQ8EgI0VKLZ8yrxSlsha1g3DZpxaKU/dr_pourreza/HRN/models/hrn.py", line 150, in predict_base
im_tensor, lm_tensor, im_hd_tensor, lm_hd_tensor, mask = self.read_data(img, landmarks, self.lm3d_std, image_res=512, img_fat=fatbgr)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/content/drive/.shortcut-targets-by-id/1ZcQQ8EgI0VKLZ8yrxSlsha1g3DZpxaKU/dr_pourreza/HRN/models/hrn.py", line 68, in read_data
mask_lr = self.face_sess.run(self.face_sess.graph.get_tensor_by_name('output_alpha:0'), feed_dict={'input_image:0': np.array(im_lr)})
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.11/dist-packages/tensorflow/python/client/session.py", line 977, in run
result = self._run(None, fetches, feed_dict, options_ptr,
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.11/dist-packages/tensorflow/python/client/session.py", line 1220, in _run
results = self._do_run(handle, final_targets, final_fetches,
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.11/dist-packages/tensorflow/python/client/session.py", line 1400, in _do_run
return self._do_call(_run_fn, feeds, fetches, targets, options,
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.11/dist-packages/tensorflow/python/client/session.py", line 1426, in _do_call
raise type(e)(node_def, op, message) # pylint: disable=no-value-for-parameter
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
tensorflow.python.framework.errors_impl.InvalidArgumentError: Graph execution error:

Detected at node 'coarse/Conv/Conv2D' defined at (most recent call last):
Node: 'coarse/Conv/Conv2D'
Detected at node 'coarse/Conv/Conv2D' defined at (most recent call last):
Node: 'coarse/Conv/Conv2D'
2 root error(s) found.
(0) INVALID_ARGUMENT: No DNN in stream executor.
[[{{node coarse/Conv/Conv2D}}]]
[[output_alpha/_3]]
(1) INVALID_ARGUMENT: No DNN in stream executor.
[[{{node coarse/Conv/Conv2D}}]]
0 successful operations.
0 derived errors ignored.

Original stack trace for 'coarse/Conv/Conv2D':

unfortunately this happened in google colab with cuda 12.5 and tensorflow version 2.18:(