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
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)
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)