Skip to content

import ai_edge_torch - NotFoundError: libtfkernel_sobol_op.so: undefined symbol #874

@gael-gitgit

Description

@gael-gitgit

Description of the bug:

I followed notebook in doc/getting started on google collab.

when installing with :
!pip install -r https://raw.githubusercontent.com/google-ai-edge/ai-edge-torch/main/requirements.txt

I got :
ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.
grpcio-status 1.71.2 requires protobuf<6.0dev,>=5.26.1, but you have protobuf 6.33.2 which is incompatible.
google-ai-generativelanguage 0.6.15 requires protobuf!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<6.0.0dev,>=3.20.2, but you have protobuf 6.33.2 which is incompatible.
tensorflow 2.19.0 requires protobuf!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<6.0.0dev,>=3.20.3, but you have protobuf 6.33.2 which is incompatible.
Successfully installed ai-edge-litert-2.1.0 ai-edge-litert-nightly-2.2.0.dev20251217 ai-edge-quantizer-nightly-0.5.0.dev20251218 backports.strenum-1.2.8 h5py-3.14.0 keras-nightly-3.12.0.dev2025100703 protobuf-6.33.2 tb-nightly-2.20.0a20250717 tf-nightly-2.21.0.dev20251218 torch-xla2-0.0.1.dev202412041639
WARNING: The following packages were previously imported in this runtime:
[google]
You must restart the runtime in order to use newly installed versions.

After restart session , when i try to import :
import ai_edge_torch

I got :

NotFoundError Traceback (most recent call last)
/tmp/ipython-input-4035273754.py in <cell line: 0>()
1 import numpy as np
----> 2 import ai_edge_torch
3 import torch
4 import torchvision

4 frames
/usr/local/lib/python3.12/dist-packages/tensorflow/python/framework/load_library.py in load_library(library_location)
149
150 for lib in kernel_libraries:
--> 151 py_tf.TF_LoadLibrary(lib)
152
153 else:

NotFoundError: /usr/local/lib/python3.12/dist-packages/tensorflow/core/kernels/libtfkernel_sobol_op.so: undefined symbol: _ZN10tensorflow15TensorShapeBaseINS_11TensorShapeEEC2EN4absl12lts_202308024SpanIKlEE

Actual vs expected behavior:

No response

Any other information you'd like to share?

No response

Metadata

Metadata

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions