赞
踩
进行旋转目标检测代码训练过程中,需对nms_rotated进行编译
进入nms_rotated文件夹,输入编译指令
- cd utils/nms_rotated
- python setup.py develop
记录实现过程中出现的错误及解决方法
配置环境:
python 3.9,torch==1.10.1+cu113,torchvision==0.11.2+cu113,torchaudio==0.10.1+cu113
网上给出的大部分解决方法:
export CUDA_HOME=/usr/local/cuda
应该是有用的,再次运行出现了下面这个错误
这个问题最后没有去解决,直接重新配置了一个环境
配置环境:
python 3.8,
pytorch==1.10.0+cu11.3,torchvision==0.11.0+cu11.3,torchaudio==0.10.0+cu11.3
出现下面这个问题
根据黄色框里的内容,应该是gcc版本太高了,降低gcc版本
查看gcc、g++版本指令:
- ls /usr/bin/g++*
- ls /usr/bin/gcc*
安装gcc、g++指令:
- sudo apt-get install gcc-9
- sudo apt-get install g++-9
设置gcc、g++优先级,最后的数字代表优先程度,数值越大,优先程度越高
- sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 100
- sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-9 100
运行上述指令后会分别出现:
- update-alternatives: 使用 /usr/bin/gcc-9 来在自动模式中提供 /usr/bin/gcc (gcc)
- update-alternatives: 使用 /usr/bin/g++-9 来在自动模式中提供 /usr/bin/g++ (g++)
再次运行,成功编译
python setup.py develop
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。