在ubuntu22.04 python3.10 pytorch2.2下运行编译votenet中的pointnet++

文章讲述了在FacebookResearch的Votenet项目中遇到的编译问题,涉及到对Tensor数据指针的处理。作者提供了针对较低torch版本的修改建议,包括将`data`改为`data_ptr`,以及简化`new_xyz`方法的检查。同时,作者分享了已修改的代码仓库供他人参考。
摘要由CSDN通过智能技术生成

 Unable to compile pointnet2 · Issue #108 · facebookresearch/votenet (github.com)

在这个issue中以及有人给出了较低torch版本的编译解决方案,但是最新的pytorch还需要进行一些代码修改。

修改如下:

错误1如上图,我们需要修改所有.cpp文件中涉及向cuda kernel传入Tensor数据指针的代码。

上图中有两处错误处,除了data<T>改为data_ptr<T>这处错误外,还有new_xyz(类型为at::Tensor)不需要.type().is_cuda(),直接.is_cuda()即可。

如果你懒得修改这么多处代码,我这里将我修改好的代码上传到github了,大家可以按需拿取

代码仓库

  • 8
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值