ThibaultGROUEIX / ChamferDistancePytorch

Chamfer Distance in Pytorch with f-score
MIT License
336 stars 45 forks source link

CUDA ERROR #27

Open Pomato1999 opened 2 years ago

Pomato1999 commented 2 years ago

" no kernel image is available for execution on the device !" but when I train other net ,this did not happen. so I think if there is a problem with the installation.

nicolasugrinovic commented 2 years ago

Did you solve this? Are you still having that error message? I get something similar when the code is executed:

error in nnd updateOutput: no kernel image is available for execution on the device error in nnd get grad: no kernel image is available for execution on the device

Possibly something to do with the CUDA and nvidia driver versions?

Pomato1999 commented 2 years ago

Did you solve this? Are you still having that error message? I get something similar when the code is executed:

error in nnd updateOutput: no kernel image is available for execution on the device error in nnd get grad: no kernel image is available for execution on the device

Possibly something to do with the CUDA and nvidia driver versions?

did not . When I execute the test code, import torch, chamfer3D.dist_chamfer_3D, fscore chamLoss = chamfer3D.dist_chamfer_3D.chamfer_3DDist() points1 = torch.rand(32, 1000, 3).cuda() points2 = torch.rand(32, 2000, 3, requires_grad=True).cuda() All the above are normal,but, ‘ dist1, dist2, idx1, idx2 = chamLoss(points1, points2) ’, ' no kernel image is available for execution on the device!‘ It bothers me,I don't know what went wrong. Your situation is the same as mine,so Which version do you use ?