Issue with Loading TorchScript in Rhino/Grasshopper

I’m encountering an error when trying to load a “TorchScript model” inside Rhino/Grasshopper using RhinoCode. The script I’m using is:

model = torch.jit.load(model_path)

However, I get the following error:

Unknown builtin op: torch_scatter::segment_sum_csr.
Could not find any similar ops to torch_scatter::segment_sum_csr.

This seems related to “torch_scatter”, but my code runs fine in PyCharm with the same Python environment. It looks like there might be an issue with installing “torch_scatter” properly in Rhino.

Has anyone encountered this issue when using PyTorch Geometric inside Rhino/Grasshopper? Any suggestions on how to resolve this?

Thanks in advance!

1 post - 1 participant

Read full topic