Minimal example not working in colab

[FilippoBoni1921]

Description of the bug:

Here there is the code

!pip install -r https://raw.githubusercontent.com/google-ai-edge/ai-edge-torch/main/requirements.txt
!pip install ai-edge-torch-nightly
import numpy as np
import ai_edge_torch
import torch
import torchvision

Error:

---------------------------------------------------------------------------
NotFoundError                             Traceback (most recent call last)
[/tmp/ipython-input-2940186061.py](https://localhost:8080/#) 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](https://localhost:8080/#) 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

[guillem-ms]

any updates on this?

[gaikwadrahul8]

Hi, @FilippoBoni1921 and @guillem-ms
I apologize for the delay in my response and I was able to reproduce the same behavior from my end which mentioned in the issue template in the Google colab but when I uninstalled TensorFlow then things are working as expected even I tried with sample example please refer this gist-file as reference

I have verified this using a sample example. Please refer to this Gist for the working reproduction:https://colab.sandbox.google.com/gist/gaikwadrahul8/a2e93898bce1c069a0755ff4ce7b397f/ai-edge-torch-859.ipynb

I would request you to please give it try from your end and let me know is it working as expected after uninstalling TensorFlow

Thank you for your cooperation and patience.

[guillem-ms]

Hi @gaikwadrahul8 !
The provided example works, it also solved an inconvenient issue I had on my own enviroment. Thanks.

However, It would be nice that you modify the minimal example and explain somewhere that this library is incompatible with Tensorflow.