运行作者源码
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模式运行的话,需要特定的显卡型