InternRobotics / InternRobotics/MeshCoder

Torch Scatter with Pytorch 2.4.1 and Cu118

Open
#6 2 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Jupyter Notebook
Stars
534
Forks
29
PR merge metrics
No merged PRs in 30d

Description

Hey Team,

I have been trying to install the MeshCoder for inference and for some reason torch scatter does not have the supported version on torch 2.4.1 and Cuda 11.8. I am wondering how did you guys manage to make it work as I am getting errror by using the given pip install torch-scatter -f https://data.pyg.org/whl/torch-2.4.1%2Bcu118.html, as you look closely while installing, the screenshot shows that torch-scatter uses torch 2.4.0 as the cached version.

Image

Could you help me figure my way out?

The error is

Traceback (most recent call last):
  File "/mnt/o/share/Tathya/MeshCoder/recipes/inference/local_inference/custom_inference/inference_shape2code_point_input.py", line 20, in <module>
    from llama_recipes.custom_models.modules.shape_tokenizer import ShapeTokenizer
  File "/mnt/o/share/Tathya/MeshCoder/src/llama_recipes/custom_models/modules/shape_tokenizer.py", line 1, in <module>
    from llama_recipes.custom_models.modules.pcd_proj.point_projection import PointTriplaneProjector
  File "/mnt/o/share/Tathya/MeshCoder/src/llama_recipes/custom_models/modules/pcd_proj/point_projection.py", line 4, in <module>
    from torch_scatter import scatter, scatter_mean, scatter_max
  File "/home/maint/miniforge3/envs/meshcoder/lib/python3.10/site-packages/torch_scatter/__init__.py", line 16, in <module>
    torch.ops.load_library(spec.origin)
  File "/home/maint/miniforge3/envs/meshcoder/lib/python3.10/site-packages/torch/_ops.py", line 1295, in load_library
    ctypes.CDLL(path)
  File "/home/maint/miniforge3/envs/meshcoder/lib/python3.10/ctypes/__init__.py", line 374, in __init__
    self._handle = _dlopen(self._name, mode)
OSError: /home/maint/miniforge3/envs/meshcoder/lib/python3.10/site-packages/torch_scatter/_version_cuda.so: undefined symbol: _ZN5torch3jit17parseSchemaOrNameERKSsb```

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start with the torch-scatter installation command and the import in src/llama_recipes/custom_models/modules/pcd_proj/point_projection.py, then reproduce the failure by running recipes/inference/local_inference/custom_inference/inference_shape2code_point_input.py. Check the installed PyTorch and torch-scatter versions against the reported undefined-symbol error; done means torch_scatter imports successfully and the inference entry point proceeds past startup.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
build-system, machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.