OSError: /home/yc/anaconda3/envs/mask3d/lib/python3.10/site-packages/torch_scatter/_version_cuda.so: undefined symbol: _ZN3c1017RegisterOperatorsD1Ev

[BrsCchang]

~/workspace/Mask3D-main$ python main_instance_segmentation.py general.experiment_name="validation" general.project_name="stpls3d" data/datasets=stpls3d general.num_targets=15 data.num_labels=15 data.voxel_size=0.333 data.num_workers=10 data.cache_data=true data.cropping_v1=false general.reps_per_epoch=100 model.num_queries=${CURR_QUERY} general.on_crops=true model.config.backbone.target=models.Res16UNet18B data.crop_length=${CURR_SIZE} general.eval_inner_core=50.0
Traceback (most recent call last):
File "/mnt/h/workspace/Mask3D-main/main_instance_segmentation.py", line 8, in
from trainer.trainer import InstanceSegmentation, RegularCheckpointing
File "/mnt/h/workspace/Mask3D-main/trainer/trainer.py", line 9, in
from torch_scatter import scatter_mean
File "/home/yc/anaconda3/envs/mask3d/lib/python3.10/site-packages/torch_scatter/init.py", line 16, in
torch.ops.load_library(spec.origin)
File "/home/yc/anaconda3/envs/mask3d/lib/python3.10/site-packages/torch/_ops.py", line 1295, in load_library
ctypes.CDLL(path)
File "/home/yc/anaconda3/envs/mask3d/lib/python3.10/ctypes/init.py", line 374, in init
self._handle = _dlopen(self._name, mode)
OSError: /home/yc/anaconda3/envs/mask3d/lib/python3.10/site-packages/torch_scatter/_version_cuda.so: undefined symbol: _ZN3c1017RegisterOperatorsD1Ev