pointnet的环境配置见我上一篇博文源碼见。刚开始跑的时候通过语句:
禁用cudnn,只用CPU跑确实是可以跑起来的。但是如果用CUDA则报错“CUDA_STATUS_NOT_SUPPORTED”,这是版本不匹配所致在我上一篇博文中已经解决。
可是还是有“cuda out of memory”的错误(如下图)我将代码中的batch_size从24改成了8,总算是可以跑起来了大概跑完modelnet40的分类任务需要66个小时。
根据参考资料我使用命令nvidia-smi查看gpu占用情况,如图所示:
不过老板说这个无伤大雅显存不够以后买新显卡,那我暂时就不管了吧~
60多个小时の后看结果就是了
运行的时候出现了一个小错误(下图),想要导入pointnet.py这个模块结果IDE想给我导入别的文件夹下的名为pointnet的包。