'Segmentation fault (core dumped)' when moving a tensor to a device

ThibaultGROUEIX / ChamferDistancePytorch

Chamfer Distance in Pytorch with f-score

MIT License

326 stars 43 forks source link

'Segmentation fault (core dumped)' when moving a tensor to a device #21

Open qihao067 opened 2 years ago

qihao067 commented 2 years ago

Hi! Thanks for your work.

I have run python setup.py install in chamfer2D with no error. However, when I am using this package in Python 3.6, I got the following problem:

>>> import torch >>> import chamfer_2D >>> torch.tensor(1,device='cuda') Segmentation fault (core dumped)

If I don't import chamfer_2D, everything works fine.

I am using CUDA 10.1 and PyTorch 1.4.0

Do you have any idea on how to solve this problem?

Best regards

ThibaultGROUEIX commented 2 years ago

Hi! This looks serious ! Thanks for the minimal XP. I don't know the fix but would gladly add it to the repo if you find it. I suspect it has to do with the latest torch/Vida release. I'll run some tests on my end.