mmdetection3D RuntimeError: Error compiling objects for extension
最近在配置open-mmlab的mmdetection3d项目,其中大部分的步骤都可以很方便的完成,但是在CUDA版本和torch版本上会有一定的问题,详细记录下:前面的几个步骤都没什么问题,就最后一步的python setup develop的时候,出现了RuntimeError: Error compiling objects for extension 这个错误https://github.
最近在配置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环节
更多推荐
所有评论(0)