FarthestPointSimple issue when importing meta graph

[shenyxcode]

Issue type

Bug

Have you reproduced the bug with TensorFlow Nightly?

No

Source

source

TensorFlow version

tf 1.14

Custom code

Yes

OS platform and distribution

Ubuntu 16.04

Mobile device

No response

Python version

3.6

Bazel version

No response

GCC/compiler version

No response

CUDA/cuDNN version

cuda 11.1

GPU model and memory

No response

Current behavior?

when loading the pretrained model 'R-PointNet' from repository https://github.com/ericyi/GSPN ,it has the problem when running code saver = tf.compat.v1.train.import_meta_graph('gspn/GSPN-master/models/model.ckpt.meta', clear_devices=True),
I've tried to switch several tf versions including 1.x and 2.x ,but still have the similar problem, when using tf 1.14 the traceback is KeyError, full traceback is provided below

Standalone code to reproduce the issue

import tensorflow as tf
tf.compat.v1.disable_eager_execution()
tf.test.is_gpu_available()
tf.test.is_built_with_cuda()
saver = tf.compat.v1.train.import_meta_graph('gspn/GSPN-master/models/model.ckpt.meta', clear_devices=True)
# for op in tf.compat.v1.get_default_graph().get_operations():
#     print(op.name)

Relevant log output

/home/lingyr/miniconda3/bin/conda run -n gspn --no-capture-output python /home/lingyr/hongzhan/22_1_GetModelTensorName.py 
/home/lingyr/miniconda3/envs/gspn/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py:516: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'.
  _np_qint8 = np.dtype([("qint8", np.int8, 1)])
/home/lingyr/miniconda3/envs/gspn/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py:517: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'.
  _np_quint8 = np.dtype([("quint8", np.uint8, 1)])
/home/lingyr/miniconda3/envs/gspn/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py:518: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'.
  _np_qint16 = np.dtype([("qint16", np.int16, 1)])
/home/lingyr/miniconda3/envs/gspn/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py:519: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'.
  _np_quint16 = np.dtype([("quint16", np.uint16, 1)])
/home/lingyr/miniconda3/envs/gspn/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py:520: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'.
  _np_qint32 = np.dtype([("qint32", np.int32, 1)])
/home/lingyr/miniconda3/envs/gspn/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py:525: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'.
  np_resource = np.dtype([("resource", np.ubyte, 1)])
/home/lingyr/miniconda3/envs/gspn/lib/python3.6/site-packages/tensorboard/compat/tensorflow_stub/dtypes.py:541: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'.
  _np_qint8 = np.dtype([("qint8", np.int8, 1)])
/home/lingyr/miniconda3/envs/gspn/lib/python3.6/site-packages/tensorboard/compat/tensorflow_stub/dtypes.py:542: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'.
  _np_quint8 = np.dtype([("quint8", np.uint8, 1)])
/home/lingyr/miniconda3/envs/gspn/lib/python3.6/site-packages/tensorboard/compat/tensorflow_stub/dtypes.py:543: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'.
  _np_qint16 = np.dtype([("qint16", np.int16, 1)])
/home/lingyr/miniconda3/envs/gspn/lib/python3.6/site-packages/tensorboard/compat/tensorflow_stub/dtypes.py:544: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'.
  _np_quint16 = np.dtype([("quint16", np.uint16, 1)])
/home/lingyr/miniconda3/envs/gspn/lib/python3.6/site-packages/tensorboard/compat/tensorflow_stub/dtypes.py:545: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'.
  _np_qint32 = np.dtype([("qint32", np.int32, 1)])
/home/lingyr/miniconda3/envs/gspn/lib/python3.6/site-packages/tensorboard/compat/tensorflow_stub/dtypes.py:550: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'.
  np_resource = np.dtype([("resource", np.ubyte, 1)])
2024-04-30 20:36:20.508145: I tensorflow/core/platform/cpu_feature_guard.cc:142] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA
2024-04-30 20:36:20.534394: I tensorflow/stream_executor/platform/default/dso_loader.cc:42] Successfully opened dynamic library libcuda.so.1
2024-04-30 20:36:21.354329: I tensorflow/compiler/xla/service/service.cc:168] XLA service 0x55dc1c2b0870 executing computations on platform CUDA. Devices:
2024-04-30 20:36:21.354405: I tensorflow/compiler/xla/service/service.cc:175]   StreamExecutor device (0): NVIDIA GeForce RTX 3090, Compute Capability 8.6
2024-04-30 20:36:21.354426: I tensorflow/compiler/xla/service/service.cc:175]   StreamExecutor device (1): NVIDIA GeForce RTX 3090, Compute Capability 8.6
2024-04-30 20:36:21.354443: I tensorflow/compiler/xla/service/service.cc:175]   StreamExecutor device (2): NVIDIA GeForce RTX 3090, Compute Capability 8.6
2024-04-30 20:36:21.354462: I tensorflow/compiler/xla/service/service.cc:175]   StreamExecutor device (3): NVIDIA GeForce RTX 3090, Compute Capability 8.6
2024-04-30 20:36:21.358802: I tensorflow/core/platform/profile_utils/cpu_utils.cc:94] CPU Frequency: 2499865000 Hz
2024-04-30 20:36:21.361254: I tensorflow/compiler/xla/service/service.cc:168] XLA service 0x55dc1a4cb470 executing computations on platform Host. Devices:
2024-04-30 20:36:21.361374: I tensorflow/compiler/xla/service/service.cc:175]   StreamExecutor device (0): <undefined>, <undefined>
2024-04-30 20:36:21.362114: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1640] Found device 0 with properties: 
name: NVIDIA GeForce RTX 3090 major: 8 minor: 6 memoryClockRate(GHz): 1.695
pciBusID: 0000:08:00.0
2024-04-30 20:36:21.362895: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1640] Found device 1 with properties: 
name: NVIDIA GeForce RTX 3090 major: 8 minor: 6 memoryClockRate(GHz): 1.695
pciBusID: 0000:09:00.0
2024-04-30 20:36:21.363591: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1640] Found device 2 with properties: 
name: NVIDIA GeForce RTX 3090 major: 8 minor: 6 memoryClockRate(GHz): 1.695
pciBusID: 0000:88:00.0
2024-04-30 20:36:21.364237: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1640] Found device 3 with properties: 
name: NVIDIA GeForce RTX 3090 major: 8 minor: 6 memoryClockRate(GHz): 1.695
pciBusID: 0000:89:00.0
2024-04-30 20:36:21.364614: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Could not dlopen library 'libcudart.so.10.0'; dlerror: libcudart.so.10.0: cannot open shared object file: No such file or directory
2024-04-30 20:36:21.364758: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Could not dlopen library 'libcublas.so.10.0'; dlerror: libcublas.so.10.0: cannot open shared object file: No such file or directory
2024-04-30 20:36:21.367031: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Could not dlopen library 'libcufft.so.10.0'; dlerror: libcufft.so.10.0: cannot open shared object file: No such file or directory
2024-04-30 20:36:21.367153: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Could not dlopen library 'libcurand.so.10.0'; dlerror: libcurand.so.10.0: cannot open shared object file: No such file or directory
2024-04-30 20:36:21.367257: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Could not dlopen library 'libcusolver.so.10.0'; dlerror: libcusolver.so.10.0: cannot open shared object file: No such file or directory
2024-04-30 20:36:21.367359: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Could not dlopen library 'libcusparse.so.10.0'; dlerror: libcusparse.so.10.0: cannot open shared object file: No such file or directory
2024-04-30 20:36:21.367464: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Could not dlopen library 'libcudnn.so.7'; dlerror: libcudnn.so.7: cannot open shared object file: No such file or directory
2024-04-30 20:36:21.367489: W tensorflow/core/common_runtime/gpu/gpu_device.cc:1663] Cannot dlopen some GPU libraries. Skipping registering GPU devices...
2024-04-30 20:36:21.367536: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1181] Device interconnect StreamExecutor with strength 1 edge matrix:
2024-04-30 20:36:21.367556: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1187]      0 1 2 3 
2024-04-30 20:36:21.367573: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1200] 0:   N N N N 
2024-04-30 20:36:21.367588: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1200] 1:   N N N N 
2024-04-30 20:36:21.367602: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1200] 2:   N N N N 
2024-04-30 20:36:21.367617: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1200] 3:   N N N N 
Traceback (most recent call last):
  File "/home/lingyr/hongzhan/22_1_GetModelTensorName.py", line 5, in <module>
    saver = tf.compat.v1.train.import_meta_graph('gspn/GSPN-master/models/model.ckpt.meta', clear_devices=True)
  File "/home/lingyr/miniconda3/envs/gspn/lib/python3.6/site-packages/tensorflow/python/training/saver.py", line 1449, in import_meta_graph
    **kwargs)[0]
  File "/home/lingyr/miniconda3/envs/gspn/lib/python3.6/site-packages/tensorflow/python/training/saver.py", line 1473, in _import_meta_graph_with_return_elements
    **kwargs))
  File "/home/lingyr/miniconda3/envs/gspn/lib/python3.6/site-packages/tensorflow/python/framework/meta_graph.py", line 857, in import_scoped_meta_graph_with_return_elements
    return_elements=return_elements)
  File "/home/lingyr/miniconda3/envs/gspn/lib/python3.6/site-packages/tensorflow/python/util/deprecation.py", line 507, in new_func
    return func(*args, **kwargs)
  File "/home/lingyr/miniconda3/envs/gspn/lib/python3.6/site-packages/tensorflow/python/framework/importer.py", line 400, in import_graph_def
    _RemoveDefaultAttrs(op_dict, producer_op_list, graph_def)
  File "/home/lingyr/miniconda3/envs/gspn/lib/python3.6/site-packages/tensorflow/python/framework/importer.py", line 160, in _RemoveDefaultAttrs
    op_def = op_dict[node.op]
KeyError: 'FarthestPointSample'
ERROR conda.cli.main_run:execute(124): `conda run python /home/lingyr/hongzhan/22_1_GetModelTensorName.py` failed. (See above for error)

Process finished with exit code 1

[google-ml-butler]

Are you satisfied with the resolution of your issue?
Yes
No