InternRobotics / InternRobotics/MeshCoder
Torch Scatter with Pytorch 2.4.1 and Cu118
Nadie ha tomado este issue todavía.
- Lenguaje dominante
- Jupyter Notebook
- Estrellas
- 534
- Forks
- 29
- Métricas de merge de PR
- Sin PR fusionados en 30 d
Descripción
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```
Guía de contribución
No hay ninguna guía de contribución indexada para este repositorio
Primeros pasos
- Lee el issue completo y luego la guía de contribución del proyecto.
- Comenta en el issue que vas a ocuparte — evita que dos personas hagan lo mismo.
- Haz un fork del repositorio y trabaja en una rama.
- Abre un pull request que haga referencia al número del issue.
Línea de trabajo
Comienza con el comando de instalación de torch-scatter y el import en src/llama_recipes/custom_models/modules/pcd_proj/point_projection.py; después reproduce el fallo ejecutando recipes/inference/local_inference/custom_inference/inference_shape2code_point_input.py. Comprueba las versiones instaladas de PyTorch y torch-scatter con respecto al error reportado de símbolo no definido; se considera terminado cuando torch_scatter se importa correctamente y el punto de entrada de inferencia avanza más allá del inicio.
Escrito por el modelo de indexación a partir del texto del issue.
Evaluación
- Stack tecnológico
- python, pytorch
- Área
- build-system, machine-learning
- Tipo de issue
- Error
- Dificultad
- 4/5
- Tiempo estimado
- 3-5 días
- Estado de actividad
- Estancado
- Claridad
- Necesita aclaración
- Aptitud para principiantes
- 25/100