InternRobotics / InternRobotics/MeshCoder

Torch Scatter with Pytorch 2.4.1 and Cu118

Aperta
#6 2 commenti 0 reazioni 0 assegnatari Vedi su GitHub

Nessuno ha ancora preso questa issue.

Lingua principale
Jupyter Notebook
Stelle
534
Fork
29
Metriche di merge delle PR
Nessuna PR unita negli ultimi 30g

Descrizione

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

Guida per i contributori

Nessuna guida per i contributori indicizzata per questo repository

Come iniziare

  1. Leggi tutta la issue e poi la guida ai contributi del progetto.
  2. Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
  3. Fai un fork del repository e lavora su un branch.
  4. Apri una pull request che faccia riferimento al numero della issue.

Direzione di ricerca

Inizia con il comando di installazione di torch-scatter e l’import in src/llama_recipes/custom_models/modules/pcd_proj/point_projection.py, quindi riproduci il fallimento eseguendo recipes/inference/local_inference/custom_inference/inference_shape2code_point_input.py. Verifica le versioni installate di PyTorch e torch-scatter rispetto all’errore segnalato relativo a un simbolo non definito; il lavoro è completato quando torch_scatter viene importato correttamente e il punto di ingresso dell’inferenza procede oltre l’avvio.

Scritto dal modello di indicizzazione a partire dal testo della issue.

Valutazione

Stack tecnologico
python, pytorch
Ambito
build-system, machine-learning
Tipo di issue
Bug
Difficoltà
4/5
Tempo stimato
3-5 giorni
Stato di attività
Ferma
Chiarezza
Da chiarire
Idoneità per principianti
25/100

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.