ImportError: libtorch_cuda_cu.so: cannot open shared object file: No such file or directory

[Woo-ing]

I have get an error when I run the nipynb in Colab :

!source activate hybrik && python scripts/demo_image.py --img-dir examples --out-dir res

Traceback (most recent call last):
File "scripts/demo_image.py", line 12, in
from hybrik.utils.render_pytorch3d import render_mesh
File "/content/HybrIK/hybrik/utils/render_pytorch3d.py", line 3, in
import pytorch3d.renderer
File "/usr/local/envs/hybrik/lib/python3.8/site-packages/pytorch3d/renderer/init.py", line 7, in
from .blending import (
File "/usr/local/envs/hybrik/lib/python3.8/site-packages/pytorch3d/renderer/blending.py", line 10, in
from pytorch3d import _C
ImportError: libtorch_cuda_cu.so: cannot open shared object file: No such file or directory

Can someone help me?

[alexinnovap]

This also happened to me, did you manage to fix it?

[AmoghTiwari]

The above issue is an issue with pytorch3d installation. I had also encountered it. And while I was not able to solve this directly, I tried installing things from scratch following pytorch3d's official installation instructions, and was able to solve it eventually. You can look at #186 for details on how I installed the packages.