[Colab] Error on KITTI image inference

[m1litaru]

Hello, @zachteed, @jiadeng @heilaw! Congratulations on your work!

I have been trying during the past days to run the demo with sample images in assets folder. I have tried doing this locally but, since I don't have GPU on my Linux laptop, I have switched to Google Colab. I have installed all the necessary packages and, unfortunately, while running !python scripts/demo.py --network=raft3d.raft3d --model=raft3d.pth I can't get rid of this error:

ImportError: /usr/local/lib/python3.7/dist-packages/lietorch_backends.cpython-37m-x86_64-linux-gnu.so: undefined symbol: _ZNK2at10TensorBase8data_ptrIdEEPT_v

I have been trying to solve it, I guess it is related to the environment I have configured (probably to the torch version).
Can you please specify to me the versions (of torch, torchvision and anything that matters) that suit this solution?

This is the link to the Colaboratory file I am working on.

Thank you!

P.S. @troylhy1991, I read on another issue that you were also inferring on KITTI dataset. Do you have any idea that might work?

Thank you guys for your time. I can provide you any other detail to what I am trying to do in case that might help.