WIN11安装detectron2踩坑

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

 安装步骤:

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
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值