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

Windows10+python3.6+cuda11.0+pythorch1.9.0部署Detectron2环境心得分享_第1张图片

 如图所示,经过不懈的努力终于将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组建

Windows10+python3.6+cuda11.0+pythorch1.9.0部署Detectron2环境心得分享_第2张图片

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不适配,添加系统环境变量,如图

Windows10+python3.6+cuda11.0+pythorch1.9.0部署Detectron2环境心得分享_第3张图片

TORCH_CUDA_ARCH_LIST
8.0

 

你可能感兴趣的:(工作环境配置,python,pip,conda)