-
Notifications
You must be signed in to change notification settings - Fork 50
Open
Description
Hi,
so I pip install chamferdist, installed via pip. But when I run my training loop where both source and target is in cuda.(), I get the following error:
Traceback (most recent call last):
File "train.py", line 114, in <module>
loss_net = chamferDist(pred, gt, bidirectional=True)
File "/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py", line 722, in _call_impl
result = self.forward(*input, **kwargs)
File "/opt/conda/lib/python3.7/site-packages/chamferdist/chamfer.py", line 82, in forward
K=1,
File "/opt/conda/lib/python3.7/site-packages/chamferdist/chamfer.py", line 267, in knn_points
p1, p2, lengths1, lengths2, K, version, return_sorted
File "/opt/conda/lib/python3.7/site-packages/chamferdist/chamfer.py", line 162, in forward
idx, dists = _C.knn_points_idx(p1, p2, lengths1, lengths2, K, version)
RuntimeError: Not compiled with GPU support.
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels