您好,欢迎来到外链网!
当前位置:外链网 » 站长资讯 » 专业问答 » 文章详细 订阅RssFeed

mmdetection3druntimeer

来源:互联网 浏览:74次 时间:2023-04-08

? ? ? 最近在配置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环节?

? ??

? ? ?

42695240