NVIDIA-Merlin / Merlin

NVIDIA Merlin is an open source library providing end-to-end GPU-accelerated recommender systems, from feature engineering and preprocessing to training deep learning models and running inference in production.
Apache License 2.0
758 stars 113 forks source link

Fix sok test error due to conflict version of tensorflow-metadata #1057

Closed EmmaQiaoCh closed 1 year ago

EmmaQiaoCh commented 1 year ago

Fix the sok test failure as below:

2023-07-19 08:32:42.905432: I tensorflow/core/platform/cpu_feature_guard.cc:183] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations. To enable the following instructions: SSE3 SSE4.1 SSE4.2 AVX, in other operations, rebuild TensorFlow with the appropriate compiler flags. Traceback (most recent call last): File "/hugectr/sparse_operation_kit/sparse_operation_kit/experiment/test/function_test/tf2/variable/sok_sgd_test.py", line 19, in import horovod.tensorflow as hvd File "/usr/local/lib/python3.10/dist-packages/horovod/tensorflow/init.py", line 507, in _SessionRunHook = tf.estimator.SessionRunHook File "/usr/local/lib/python3.10/dist-packages/tensorflow/python/util/lazy_loader.py", line 58, in getattr module = self._load() File "/usr/local/lib/python3.10/dist-packages/tensorflow/python/util/lazy_loader.py", line 41, in _load module = importlib.import_module(self.name) File "/usr/lib/python3.10/importlib/init.py", line 126, in import_module return _bootstrap._gcd_import(name[level:], package, level) File "/usr/local/lib/python3.10/dist-packages/tensorflow_estimator/init.py", line 8, in from tensorflow_estimator._api.v1 import estimator File "/usr/local/lib/python3.10/dist-packages/tensorflow_estimator/_api/v1/estimator/init.py", line 11, in from tensorflow_estimator._api.v1.estimator import tpu File "/usr/local/lib/python3.10/dist-packages/tensorflow_estimator/_api/v1/estimator/tpu/init.py", line 12, in from tensorflow_estimator.python.estimator.tpu.tpu_estimator import TPUEstimator File "/usr/local/lib/python3.10/dist-packages/tensorflow_estimator/python/estimator/tpu/tpu_estimator.py", line 46, in from tensorflow.python.ops import ref_variable ImportError: cannot import name 'ref_variable' from 'tensorflow.python.ops' (/usr/local/lib/python3.10/dist-packages/tensorflow/python/ops/init.py)

github-actions[bot] commented 1 year ago

Documentation preview

https://nvidia-merlin.github.io/Merlin/review/pr-1057