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

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组建

Windows10+python3.6+cuda11.0+pythorch1.9.0部署Detectron2环境心得分享
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.几个常见的问题解决攻略

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

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环境心得分享
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/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球