InternRobotics / InternRobotics/MeshCoder

Torch Scatter with Pytorch 2.4.1 and Cu118

Abierto
#6 2 comentarios 0 reacciones 0 asignados Ver en GitHub

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.

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

Guía de contribución

No hay ninguna guía de contribución indexada para este repositorio

Primeros pasos

  1. Lee el issue completo y luego la guía de contribución del proyecto.
  2. Comenta en el issue que vas a ocuparte — evita que dos personas hagan lo mismo.
  3. Haz un fork del repositorio y trabaja en una rama.
  4. 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

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.