diff --git a/dali_tf_plugin/dali_tf_plugin_install_tool.py b/dali_tf_plugin/dali_tf_plugin_install_tool.py index 9c09c2e4f3..b4a3078f2c 100644 --- a/dali_tf_plugin/dali_tf_plugin_install_tool.py +++ b/dali_tf_plugin/dali_tf_plugin_install_tool.py @@ -366,7 +366,7 @@ def build(self): lib_path = os.path.join(self.plugin_dest_dir, lib_filename) # for a newer TF we need to compiler with C++17 - cpp_ver = "--std=c++14" if self.tf_version < Version("2.10") else "--std=c++17" + cpp_ver = "--std=c++14" if Version(self.tf_version) < Version("2.10") else "--std=c++17" # Note: DNDEBUG flag is needed due to issue with TensorFlow custom ops: # https://github.com/tensorflow/tensorflow/issues/17316 # Do not remove it.