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

Pointnet2_pytorch代码复现相关推荐

【神经网络】(17) EfficientNet 代码复现，网络解析，附Tensorflow完整代码
各位同学好,今天和大家分享一下如何使用 Tensorflow 复现 EfficientNet 卷积神经网络模型. EfficientNet 的网络结构和 MobileNetV3 比较相似,建议大家在学 ...
【神经网络】(16) MobileNetV3 代码复现，网络解析，附Tensorflow完整代码
各位同学好,今天和大家分享一下如何使用 Tensorflow 构建 MobileNetV3 轻量化网络模型. MobileNetV3 做了如下改动(1)更新了V2中的逆转残差结构:(2)使用NAS搜索 ...
【神经网络】(15) Xception 代码复现，网络解析，附Tensorflow完整代码
各位同学好,今天和大家分享一下如何使用 Tensorflow 构建 Xception 神经网络模型. 在前面章节中,我已经介绍了很多种轻量化卷积神经网络模型,感兴趣的可以看一下:https://blo ...
【神经网络】(14) MnasNet 代码复现，网络解析，附Tensorflow完整代码
各位同学好,今天和大家分享一下如何使用 Tensorflow 复现谷歌轻量化神经网络 MnasNet 通常而言,移动端(手机)和终端(安防监控.无人驾驶)上的设备计算能力有限,无法搭载庞大的神经网络 ...
【神经网络】(13) ShuffleNetV2 代码复现，网络解析，附Tensorflow完整代码
各位同学好,今天和大家分享一下如何使用 Tensorflow 复现轻量化神经网络 ShuffleNetV2. 为了能将神经网络模型用于移动端(手机)和终端(安防监控.无人驾驶)的实时计算,通常这些设备 ...
【神经网络】(11) 轻量化网络MobileNetV1代码复现、解析，附Tensorflow完整代码
各位同学好,今天和大家分享一下如何使用 Tensorflow 复现轻量化神经网络模型 MobileNetV1.为了能将神经网络模型用于移动端(手机)和终端(安防监控.无人驾驶)的实时计算,通常这些设备 ...
进阶必备：CNN经典论文代码复现 | 附下载链接
经常会看到类似的广告<面试算法岗,你被要求复现论文了吗?>不好意思,我真的被问过这个问题.当然也不是所有面试官都会问,究其原因,其实也很好理解.企业肯定是希望自己的产品是有竞争力,有卖点的 ...
DL之YoloV3：Yolo V3算法的简介(论文介绍)、各种DL框架代码复现、架构详解、案例应用等配图集合之详细攻略
DL之YoloV3:Yolo V3算法的简介(论文介绍).各种DL框架代码复现.架构详解.案例应用等配图集合之详细攻略目录 Yolo V3算法的简介(论文介绍) 0.YoloV3实验结果 1.Yol ...
时空上下文视觉跟踪（STC）算法的解读与代码复现
时空上下文视觉跟踪(STC)算法的解读与代码复现 zouxy09@qq.com http://blog.csdn.net/zouxy09 本博文主要是关注一篇视觉跟踪的论文.这篇论文是Kaihua Z ...

Pointnet2_pytorch代码复现

Pointnet2_pytorch代码复现相关推荐

最新文章

热门文章