laiyinping的专栏

本文针对PointNet++强大的三维点云分类功能，详细讲解怎么训练自己的数据集，在此之前，需要确保已经能够跑通源码的训练和测试，如果没有，请参考。

部件分割数据shapenetcore_partanno_segmentation_benchmark_v0_normal下载。语义分割数据Stanford3dDataset_v1.2_Aligned_Version下载。如果在Pycharm中直接运行py文件，设置参数的话，需要在IDE中进行，如下图所示。进入Edit Configurations之后，填写参数，每个参数用空格分开。任何的CUDA报错，都可以减小batch_size来解决。用Pycharm打开对应工程目录，运行一下。也可以从百度网盘下载。

错误信息。

会发现我们刚刚pip安装的是3.7.0，但是pycharm中的是3.6.2，所以需要在pycharm中再次安装。等待安装完成，在pycharm中运行导入matplotlib的文件，就不会报错啦。首先，我们在anaconda命令窗口中安装matplotlib，使用pip命令。双击matplotlib，进入到如下界面，点击Install Pacekage。我们首先查看python解释器里面matplotlib的版本。这时候我们在Pythorm中写一段代码。

打开pycharm，依次进入file——settings——Python Interpreter,如下图所示，点击右边的下拉箭头，再点击Show All，然后进行配置。回到settings界面，就可以看到已经选择的python解释器相关信息。依次进行下面的步骤，将自己的python.exe目录选择进来，点击OK。

可以选择当前的，也可以选择历史版本，根据自己的CUDA版本来定。进入帮助，选择系统信息里面的组件，查看自己的显卡的驱动版本。将解压好的cudnn三个文件，拷贝到cuda安装目录中。最后设置下环境变量，就可以啦。打开NVIDIA控制面板。

（点此即可直达Pytorch官网）复制适合自己电脑安装 Pytorch 的命令，如下图所示（我的是之前的版本，需要根据自己的CUDA版本来选择）打开 Anaconda Prompt ，先切换镜像源为国内清华镜像源，这样安装包的时候下载速度会快一些，也容易成功一些。下载Anaconda，最新版本的即可，默认安装，最好不要安装在C盘，否则后面C盘容量会很大。接下来创建一个名为pytorch的环境，并在环境中安装版本为3.9的python，命令如下。这个镜像源的配置只要一次，之后无需再配置，配置截图如下。