Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Infra] libtorch 결여 #7

Open
oMFDOo opened this issue Nov 22, 2024 · 2 comments
Open

[Infra] libtorch 결여 #7

oMFDOo opened this issue Nov 22, 2024 · 2 comments

Comments

@oMFDOo
Copy link
Owner

oMFDOo commented Nov 22, 2024

#1 과정 중 #6 해결 이후 에러

#  실행 코드 : python train_meshhead.py --config config/train_meshhead_N031.yaml
Traceback (most recent call last):
  File "train_meshhead.py", line 7, in <module>
    from lib.dataset.Dataset import MeshDataset
  File "/home/azureuser/GaussianHeadAvatar/Gaussian-Head-Avatar/lib/dataset/Dataset.py", line 12, in <module>
    from pytorch3d.renderer.cameras import look_at_view_transform
  File "/home/azureuser/anaconda3/envs/generate3d/lib/python3.8/site-packages/pytorch3d/renderer/__init__.py", line 7, in <module>
    from .blending import (
  File "/home/azureuser/anaconda3/envs/generate3d/lib/python3.8/site-packages/pytorch3d/renderer/blending.py", line 10, in <module>
    from pytorch3d import _C
ImportError: libtorch_cuda_cu.so: cannot open shared object file: No such file or directory

Image

: 해당 오류는 torch3D가 존재하지 않아 발생한 문제인데, cuda 11.8이 최대 지원 버전이므로 나는 소스코드 빌드를 해야한다.


혹시나 싶어 PyTorch3D를 다시 설치 해봤으나 올바르게 동작하지 않는다.

pip install pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py39_cu118_pyt200/download.html
@oMFDOo
Copy link
Owner Author

oMFDOo commented Nov 22, 2024

해당 문제에 대해 두 가지 방안을 세웠다.


1) Torch3D를 직접 소스코드 빌드하는 방안

 * 해당 방식은 코드 변경이 필요없어 더 간결한 방식이 예상되나 빌드의 어려움이 존재함
 * 추가적으로 KETI에서 비추천을 받은 렌더러이기에 이러한 어려움을 뚫고 빌드해야하나 고민임

2) nvdiffrast를 이용한 렌더링

 * 해당 방식은 KETI와의 방향성까지 맞추어 나갈 수 있는 좋은 방법이 될 수 있을 것 같음
 * 허나, 이런 방식에 대한 지식이 없는 상태에서 소스코드를 수정해가며 만들어 낼 수 있을 지 의문임

@oMFDOo
Copy link
Owner Author

oMFDOo commented Nov 22, 2024

[Infra] #7 PyTorch 결여 1차대안 : PyTorch3D 소스코드 빌드 #8 에서 libtorch_cuda_cu.so를 인식했음으로 1) Torch3D를 직접 소스코드 빌드하는 방안에 대해서는 해결함

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
Development

No branches or pull requests

1 participant