ChamferDistancePytorch
ChamferDistancePytorch copied to clipboard
'Segmentation fault (core dumped)' when moving a tensor to a device
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
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.