首页 Paddle框架 帖子详情
paddle自定义cuda算子问题?
收藏
快速回复
Paddle框架 其他深度学习 306 0
paddle自定义cuda算子问题?
收藏
快速回复
Paddle框架 其他深度学习 306 0

我将倒角距离的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
收藏
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户