Pointnet2_pytorch代码复现

最近在安装了一些库之后,电脑系统崩溃,之前调通的代码几乎都不能正常运行,之前一直觉得是环境版本配置问题,故重装系统与环境多次,但都不起作用,摸索很久后,代码终于能正常运行,故做此记录。


本机配置:系统:Ubuntu 18.04;CUDA:11.2;cudnn:8.1.0

                   环境:python3:3.6.8;virtualenv环境,pytorch:1.8.0


注:编译在未安装ninja的情况下进行

运行以下命令:

> cd pointenet2

> python setup.py install 

会出现类似以下的页面,其中会有错误问题的显示:

首先解决的第一个问题:

error: 'AT_CHECK' was not declaredin this scope

解决办法:pointnet2/_ext_src/src 中的所有.cpp文件以及pointnet2/_ext_src/include中的utils.h中所有的 ‘AT_CHECK’ 改为 'TORCH_CHECK',

同时,将所有的 '**.type().is_cuda()' 改为 ‘**.is_cuda()

error:Tensor.data<T> is deprecated. Please use Tensor.data_ptr<T> instead()

解决办法:将所有 .cpp 文件中,‘**.data<float/int>()’ 修改为:‘**.data_ptr<float/int>()’


修改完成后,再次运行:> python setup.py install

出现以下界面,即编译成功:

此时,在pointnet2文件夹下会出现以下文件:

随即可开始train

参考链接(如有侵权,必删):

https://github.com/erikwijmans/Pointnet2_PyTorch/commit/1d5dca2673ee2831b9b01efe597b2ba8d12726f1

  • 2
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值