如图所示,经过不懈的努力终于将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.几个常见的问题解决攻略
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
Original: https://blog.csdn.net/qq_19289445/article/details/125824815
Author: xhtnewbility
Title: Windows10+python3.6+cuda11.0+pythorch1.9.0部署Detectron2环境心得分享
原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/771815/
转载文章受原作者版权保护。转载请注明原作者出处!