When I load up the container nvcr.io/nvidia/tensorflow:20.02-tf2-py3
and run import tensorflow_addons as tfa
it fails with the following error.
/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/framework/load_library.py in load_op_library(library_filename)
55 RuntimeError: when unable to load the library or get the python wrappers.
56 """
---> 57 lib_handle = py_tf.TF_LoadLibrary(library_filename)
58 try:
59 wrappers = py_tf.GetPythonWrappers(py_tf.TF_GetOpList(lib_handle))
NotFoundError: /usr/local/lib/python3.6/dist-packages/tensorflow_addons/custom_ops/image/_distort_image_ops.so: undefined symbol: _ZN10tensorflow7strings8internal9CatPiecesB5cxx11ESt16initializer_listIN4absl11string_viewEE
It works fine in 20.01-tf2-py3
so I will just be using that for now, but thought you should know it’s crashing.