InternRobotics / InternRobotics/MeshCoder

Torch Scatter with Pytorch 2.4.1 and Cu118

未关闭
#6 2 条评论 0 个 reaction 已指派 0 人 在 GitHub 查看

还没有人认领这个 Issue。

主要语言
Jupyter Notebook
星标
534
派生
29
PR 合并指标
30 天内没有已合并 PR

描述

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```

贡献指南

这个仓库没有索引到贡献指南

从这里开始

  1. 先读完整个 Issue,再读项目的贡献指南。
  2. 在 Issue 下留言说明你要接手 —— 这能避免两个人做同样的事。
  3. Fork 仓库,在一个分支上完成修改。
  4. 提交 Pull Request,并在描述里引用这个 Issue 编号。

调研方向

先从 torch-scatter 安装命令以及 src/llama_recipes/custom_models/modules/pcd_proj/point_projection.py 中的 import 开始,然后运行 recipes/inference/local_inference/custom_inference/inference_shape2code_point_input.py 复现失败。检查已安装的 PyTorch 和 torch-scatter 版本是否与报告的未定义符号错误相符;完成的标准是 torch_scatter 成功导入,并且推理入口点继续执行而不再停留在启动阶段。

由索引模型根据 Issue 内容生成。

评估

技术栈
python, pytorch
领域
build-system, machine-learning
Issue 类型
缺陷
难度
4/5
预计耗时
3-5 天
活跃度
停滞
描述清晰度
需要澄清
新手友好度
25/100

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。