paddle自定义cuda算子问题?
收藏
我将倒角距离的torch算子改成paddle版本,但是出现下列错误:
File "unit_test.py", line 2, in
import dist_chamfer_3D
File "/home/aistudio/work/PointCloudCompletion/extentions/chamfer_dist/dist_chamfer_3D.py", line 23, in
import chamfer
ImportError: /home/aistudio/work/PointCloudCompletion/extentions/chamfer_dist/chamfer.so: undefined symbol: _Z20chamfer_cuda_forwardN6paddle6TensorES0_
pytorch出现这个错误通常是版本不对应导致,我尝试了两种环境:
- 自己的电脑 paddle 2.2.2 cuda 11.2
- AI studio paddle 2.2.2 cuda 10.1
两种环境都会出现该错误,算子编译过程没有出现错误,求大佬告知原因?
0
收藏
请登录后评论