最近在配置open-mmlab的mmdetection3d项目,其中大部分的步骤都可以很方便的完成,但是在CUDA版本和torch版本上会有一定的问题,详细记录下:

     前面的几个步骤都没什么问题,就最后一步的python setup develop的时候,出现了

       RuntimeError: Error compiling objects for extension 这个错误

      https://github.com/sshaoshuai/Pointnet2.PyTorch/issues/19 此处有一些解决办法(我自己没有尝试,感觉过于复杂,下面的解决方法要简洁许多)

     需要注意的是,根据getting_started.md中的

     

     conda 这个安装,是按照默认的CUDA10.2版本进行安装的,同时安装的版本是pytoch1.7.0版本,此时需要做出如下改动:

     conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1 -c pytorch

    此时即可按照CUDA10.1的版本对应安装,同时需要注意我在配置 bashrc里原本存在错误:

     

    第三行是有问题的,写的方式不正确,导致找不到CUDA_HOME的位置,需要修改为第四行的样式,可以正常进入setup环节 

    

     

    

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐