Zielon / MICA

MICA - Towards Metrical Reconstruction of Human Faces [ECCV2022]
https://zielon.github.io/mica/
Other
551 stars 78 forks source link

ImportError #20

Closed Jimmy142857 closed 2 years ago

Jimmy142857 commented 2 years ago

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

Jimmy142857 commented 2 years ago

// Handled~

import sys import torch pyt_version_str=torch.version.split("+")[0].replace(".", "") version_str="".join([ f"py3{sys.version_info.minor}_cu", torch.version.cuda.replace(".",""), f"_pyt{pyt_version_str}" ]) !pip install --no-index --no-cache-dir --no-deps pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html