试跑PointNet(PyTorch版本)

运行作者源码

pointnet的环境配置见我上一篇博文,源码见https://github.com/yanx27/Pointnet_Pointnet2_pytorch。刚开始跑的时候,通过语句:

torch.backends.cudnn.enabled = False

禁用cudnn,只用CPU跑,确实是可以跑起来的。但是如果用CUDA,则报错“CUDA_STATUS_NOT_SUPPORTED”,这是版本不匹配所致,在我上一篇博文中已经解决。

可是还是有“cuda out of memory”的错误(如下图)。我将代码中的batch_size从24改成了8,总算是可以跑起来了。大概跑完modelnet40的分类任务需要66个小时。

根据参考资料,我使用命令nvidia-smi查看gpu占用情况,如图所示:

不过我的GPU Memory Usage显示为N/A,按照参考资料,这是因为

当显卡在window下以WDDM模式运行时,GPU memory usage项是not available。而想要以TCC模式运行的话,需要特定的显卡型

  • 2
    点赞
  • 36
    收藏
    觉得还不错? 一键收藏
  • 10
    评论
评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值