[Python知识库]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.4187https://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

?