Windows10+python3.6+cuda11.0+pythorch1.9.0部署Detectron2环境心得分享

 如图所示,经过不懈的努力终于将Detectron2部署成功!!

1.环境准备

1.1安装Anaconda

1.2安装 Visual Studio 2019

 安装C++和python的package.

2.正式安装Detectron 2 (0.6)版本

2.1 创建Anaconda 环境

打开Ananconda prompt,创建一个新环境Detectron2,选择Python版本3.6, 当然你可以根据自己需要,换成你想要的名字即可.

conda create -n Detectron2 python=3.6

2.2 激活当前环境

conda activate Detectron2

2.3 安装Cuda组建

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

2.4 检查GPU时候可以用

打开pthon,输入以下

import torch
torch.cuda.is_available()

如果得到True,证明安装没问题。如果失败,请自行更换,cuda版本尝试

2.5 安装其他包

conda install -c anaconda cython
 
pip install opencv-python
 
pip install git+https://github.com/facebookresearch/fvcore
 
pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI
 
pip install av
 
conda install -c anaconda scipy
 
conda install -c anaconda ninja
pip install pillow matplotlib termcolor cloudpickle tabulate tensorboard termcolor tqdm yacs mock pydot wheel future

2.6从官网git或者下载到本地

https://github.com/facebookresearch/detectron2

2.7安装detectron2

cd 代码文件夹
python setup.py build develop

到这里基本上就已经安装完成了!

3.目前遇到的问题和解决办法(每个人问题不同仅分享个人看法欢迎前来讨论)

1.几个常见的问题解决攻略

https://blog.csdn.net/blink182007/article/details/124222921?ops_request_misc=&request_id=&biz_id=102&utm_term=detectron2%E5%AE%89%E8%A3%85%E6%8A%A5%E9%94%99&utm_medium=distribute.pc_search_result.none-task-blog-2~all~sobaiduweb~default-7-124222921.142^v32^control,185^v2^control&spm=1018.2226.3001.4187icon-default.png?t=M666https://blog.csdn.net/blink182007/article/details/124222921?ops_request_misc=&request_id=&biz_id=102&utm_term=detectron2%E5%AE%89%E8%A3%85%E6%8A%A5%E9%94%99&utm_medium=distribute.pc_search_result.none-task-blog-2~all~sobaiduweb~default-7-124222921.142^v32^control,185^v2^control&spm=1018.2226.3001.4187

2.接下来是我自己编译过程出现问题和解决方法

2.1依赖安装正常最后无法编译

将python版本换成3.6.x 将pytorch版本换为1.9.x

2.2nvcc fatal : Unsupported gpu architecture ‘compute_86‘

显卡算力过高与cuda不适配,添加系统环境变量,如图

TORCH_CUDA_ARCH_LIST
8.0

 

  • 2
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 7
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

xhtnewbility

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值