facebookresearch / pytorch3d

PyTorch3D is FAIR's library of reusable components for deep learning with 3D data
https://pytorch3d.org/
Other
8.81k stars 1.32k forks source link

ImportError: DLL load failed while importing _C: The specified procedure could not be found. #1872

Closed nitinmukesh closed 1 month ago

nitinmukesh commented 1 month ago

🐛 Bugs / Unexpected behaviors

I am trying to use a tool which have pytorch3d as dependency. As soon as I run the inference, following error occurs as stated in title.

Instructions To Reproduce the Issue:

Installed the following tool https://github.com/rlgnswk/NeRFFaceSpeech_Code Installed pytorch3d using wheel pip install "C:\WHEELS\pytorch3d-0.7.6+pt2.0.0cu117-cp310-cp310-win_amd64.whl"

No issues during installation.

Please include the following (depending on what the issue is):

  1. Any changes you made (git diff) or code you wrote

    No
  2. The exact command(s) you ran:

    python StyleNeRF/main_NeRFFaceSpeech_audio_driven_from_z.py --outdir=out_test --trunc=0.7 --network=pretrained_networks/ffhq_1024.pkl --test_data="test_data/test_audio/AdamSchiff_0.wav" --seeds=0
  3. What you observed (including the full logs):

    (venv) C:\ai\NeRFFaceSpeech_Code>python StyleNeRF/main_NeRFFaceSpeech_audio_driven_from_z.py --outdir=out_test --trunc=0.7 --network=pretrained_networks/ffhq_1024.pkl --test_data="test_data/test_audio/AdamSchiff_0.wav" --seeds=0
    Loading networks from "pretrained_networks/ffhq_1024.pkl"...
    Setting up PyTorch plugin "bias_act_plugin"... Done.
    C:\ai\NeRFFaceSpeech_Code\venv\lib\site-packages\torch\functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ..\aten\src\ATen\native\TensorShape.cpp:3191.)
    return _VF.meshgrid(tensors, **kwargs)  # type: ignore[attr-defined]
    Setting up PyTorch plugin "upfirdn2d_plugin"... Done.
    model [FaceReconModel] was created
    loading the model from pretrained_networks/Deep3DFaceRecon_pytorch/face_recon\epoch_20.pth
    create cuda ctx on device cuda:0
    ########################### audio2exp_model Loading Done #################################
    Traceback (most recent call last):
    File "C:\ai\NeRFFaceSpeech_Code\StyleNeRF\main_NeRFFaceSpeech_audio_driven_from_z.py", line 361, in <module>
    generate_images() # pylint: disable=no-value-for-parameter
    File "C:\ai\NeRFFaceSpeech_Code\venv\lib\site-packages\click\core.py", line 1157, in __call__
    return self.main(*args, **kwargs)
    File "C:\ai\NeRFFaceSpeech_Code\venv\lib\site-packages\click\core.py", line 1078, in main
    rv = self.invoke(ctx)
    File "C:\ai\NeRFFaceSpeech_Code\venv\lib\site-packages\click\core.py", line 1434, in invoke
    return ctx.invoke(self.callback, **ctx.params)
    File "C:\ai\NeRFFaceSpeech_Code\venv\lib\site-packages\click\core.py", line 783, in invoke
    return __callback(*args, **kwargs)
    File "C:\ai\NeRFFaceSpeech_Code\venv\lib\site-packages\click\decorators.py", line 33, in new_func
    return f(get_current_context(), *args, **kwargs)
    File "C:\ai\NeRFFaceSpeech_Code\StyleNeRF\main_NeRFFaceSpeech_audio_driven_from_z.py", line 207, in generate_images
    fitted_coeffs = fit_3dmm(outdir)
    File "C:\ai\NeRFFaceSpeech_Code\StyleNeRF\audio2NeRF_utils.py", line 127, in fit_3dmm
    from fit_single_img_custom import fit
    File "C:\ai\NeRFFaceSpeech_Code\3DMM-Fitting-Pytorch\fit_single_img_custom.py", line 2, in <module>
    from core.options import ImageFittingOptions
    File "C:\ai\NeRFFaceSpeech_Code\3DMM-Fitting-Pytorch\core\__init__.py", line 1, in <module>
    from core.BFM09Model import BFM09ReconModel
    File "C:\ai\NeRFFaceSpeech_Code\3DMM-Fitting-Pytorch\core\BFM09Model.py", line 5, in <module>
    from core.BaseModel import BaseReconModel
    File "C:\ai\NeRFFaceSpeech_Code\3DMM-Fitting-Pytorch\core\BaseModel.py", line 5, in <module>
    from pytorch3d.renderer import (
    File "C:\ai\NeRFFaceSpeech_Code\venv\lib\site-packages\pytorch3d\renderer\__init__.py", line 7, in <module>
    from .blending import (
    File "C:\ai\NeRFFaceSpeech_Code\venv\lib\site-packages\pytorch3d\renderer\blending.py", line 10, in <module>
    from pytorch3d import _C
    ImportError: DLL load failed while importing _C: The specified procedure could not be found.

Please also simplify the steps as much as possible so they do not require additional resources to run, such as a private dataset.

(venv) C:\ai\NeRFFaceSpeech_Code>pip list

Package                  Version
------------------------ ------------------
antlr4-python3-runtime   4.9.3
audioread                3.0.1
backcall                 0.2.0
certifi                  2024.8.30
cffi                     1.15.1
charset-normalizer       3.3.2
click                    8.1.7
colorama                 0.4.6
contourpy                1.3.0
cycler                   0.11.0
decorator                5.1.1
docker-pycreds           0.4.0
easydict                 1.13
einops                   0.6.1
face-alignment           1.4.1
facenet-pytorch          2.5.3
fonttools                4.38.0
fvcore                   0.1.5.post20221221
gitdb                    4.0.11
GitPython                3.1.43
idna                     3.10
imageio                  2.31.2
imhist                   0.0.4
importlib-metadata       6.7.0
imWatermark              0.0.2
invisible-watermark      0.2.0
iopath                   0.1.10
ipdb                     0.13.13
ipython                  7.34.0
jedi                     0.19.1
joblib                   1.3.2
kiwisolver               1.4.5
kornia                   0.6.12
lazy_loader              0.4
librosa                  0.10.2.post1
llvmlite                 0.39.1
lpips                    0.1.4
matplotlib               3.9.2
matplotlib-inline        0.1.7
msgpack                  1.0.5
natsort                  8.4.0
networkx                 2.6.3
ninja                    1.11.1.1
numba                    0.56.4
numpy                    1.23.5
nvdiffrast               0.3.1
nvidia-cublas-cu11       11.10.3.66
nvidia-cuda-nvrtc-cu11   11.7.99
nvidia-cuda-runtime-cu11 11.7.99
omegaconf                2.3.0
opencv-python            4.10.0.84
packaging                24.0
parso                    0.8.4
pexpect                  4.9.0
pickleshare              0.7.5
pillow                   10.4.0
pip                      22.2.1
platformdirs             4.0.0
pooch                    1.8.2
portalocker              2.10.1
prompt_toolkit           3.0.47
protobuf                 4.24.4
psutil                   6.0.0
ptyprocess               0.7.0
pycparser                2.21
Pygments                 2.17.2
pyparsing                3.1.4
python-dateutil          2.9.0.post0
pytorch3d                0.7.6+pt2.0.0cu117
PyWavelets               1.3.0
pywin32                  306
PyYAML                   6.0.2
requests                 2.31.0
safetensors              0.4.5
scikit-image             0.19.3
scikit-learn             1.0.2
scipy                    1.14.1
sentry-sdk               2.14.0
setproctitle             1.3.3
setuptools               63.2.0
six                      1.16.0
smmap                    5.0.1
soundfile                0.12.1
soxr                     0.3.7
tabulate                 0.9.0
termcolor                2.4.0
threadpoolctl            3.1.0
tifffile                 2021.11.2
tomli                    2.0.1
torch                    1.13.1+cu117
torchaudio               0.13.1+cu117
torchvision              0.14.1+cu117
tqdm                     4.66.5
traitlets                5.9.0
trimesh                  4.4.1
typing_extensions        4.12.2
urllib3                  2.0.7
wandb                    0.18.1
wcwidth                  0.2.13
wheel                    0.44.0
yacs                     0.1.8
zipp                     3.15.0
(venv) C:\ai\NeRFFaceSpeech_Code>nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Tue_May__3_19:00:59_Pacific_Daylight_Time_2022
Cuda compilation tools, release 11.7, V11.7.64
Build cuda_11.7.r11.7/compiler.31294372_0
(venv) C:\ai\NeRFFaceSpeech_Code>python --version
Python 3.10.6
bottler commented 1 month ago

You have torch version 1.13.1 but your wheel for pytorch3d looks like it was built for torch version 2. 0.0. This must match for things to work.

nitinmukesh commented 1 month ago

@bottler

Thank you.