安装步骤:
1、创建环境
conda create -n detectron2 python=3.8 -y
2、激活环境
conda activate detectron2
3、在Pytorch官网中找到对应的pytorch版本以及CUDA版本,要求pytorch版本大于1.8
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
4、安装git
conda install git
5、安装必要库
pip install cython opencv-python pillow matplotlib termcolor cloudpickle tabulate tensorboard termcolor tqdm yacs mock fvcore pydot wheel futrue -i https://pypi.tuna.tsinghua.edu.cn/simple
6、下载源码地址
https://github.com/facebookresearch/detectron2
7、cd到detectron2目录下,开始构建
python setup.py build --force develop
报错:
#error: – uns

本文详细介绍了在安装Detectron2时遇到的CUDA版本不匹配问题,包括如何找到并修改头文件,以及解决编译错误的过程。重点在于解决nms_rotated_cuda.cu中的编译错误,最后成功安装并进行预测示例。
最低0.47元/天 解锁文章
1854

被折叠的 条评论
为什么被折叠?



