detectron2安装在win10并运行测试--呕心沥血教程

最新推荐文章于 2024-07-30 16:23:33 发布

Discipline※

最新推荐文章于 2024-07-30 16:23:33 发布

阅读量5.6k

点赞数 11

分类专栏：图像处理文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/qq_51882416/article/details/122060993

版权

本文档详述了在Windows10系统中安装detectron2的全过程，包括配置CUDA、torch、VS2019，以及解决安装过程中遇到的问题，如环境设置、库的安装与验证、detectron2的下载与编译等。

摘要由CSDN通过智能技术生成

记录不易，继续加油

一、环境要求
1、Pycharm
2、anaconda自带的python3.8.8
3、cuda11.2+torch
4、vs2019
5、conda4.11.0
二、安装步骤
三、测试案例

detectron2安装在win10并运行测试

提示：以下是本篇文章正文内容，下面案例可供参考

一、确保电脑有合适的GPU——CUDA与torch版本

创建虚拟环境：
打开开始——Anaconda Prompt (anaconda)——右键，更多：以管理员身份运行；输入：conda create -n （自己取个环境名字） python=3.8.8(注意python版本号与下载的版本号一致），我这里取的名字是detectron2，即输入：conda create -n detectron2 python=3.8.8
激活虚拟环境：conda activate detectron2
检测是否激活成功：输入python，不显示warning即可
在这里插入图片描述

安装必备的库

pip/conda install numpy matplotlib
pip install opencv-python
pip install pillow cython

查看当前cuda版本：输入nvcc -V
在这里插入图片描述
如上图，cuda为10.2版本。
查看cuda和torch是否适配，输入：

python
import torch
print(torch.cuda.is_available())

在这里插入图片描述
显示为true就对了。如果是false,下面提供下载cuda和torch的方法：
cuda10.2，在cuda官网下载旧版本CUDA下载链接：

在这里插入图片描述
记住安装位置，一会cudnn安装要用！

打开路径 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin ，查看nvcc.exe

打开路径
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\CUPTI\lib64，
查看有没有cuti64_101.dll
有这个cupti64_101.dll就说明CUPT1已成功:
在这里插入图片描述
这样就下载到本地了（local）
CUDA安装步骤
接下来安装配套的cudnn7.6.5;