InternRobotics / InternRobotics/MeshCoder
Torch Scatter with Pytorch 2.4.1 and Cu118
还没有人认领这个 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.
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```
贡献指南
这个仓库没有索引到贡献指南
从这里开始
- 先读完整个 Issue,再读项目的贡献指南。
- 在 Issue 下留言说明你要接手 —— 这能避免两个人做同样的事。
- Fork 仓库,在一个分支上完成修改。
- 提交 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