Detectron2小白教程之安装试用篇

最新推荐文章于 2026-04-29 19:40:50 发布

原创

最新推荐文章于 2026-04-29 19:40:50 发布 · 3k 阅读

标签

#object detection #ai #机器学习 #图像处理 #深度学习

本文提供了一步一步的指南来在Ubuntu系统上安装Detectron2，包括安装Python、OpenCV、NVIDIA显卡驱动、CUDA11.7、PyTorch、Ninja构建系统，以及克隆和安装Detectron2库。最后，通过运行示例验证了Detectron2的安装成功。

这里写自定义目录标题

官方安装说明

Detectron2是facebook主导的支持图像分类(Image classification)，目标检测(Object detection)，语义分割（Semantic segmentation）和实例分割(instance segmentation)，全景分割(panoramic segmentation)的强大的图像识别工具。

官方安装说明

见此处

1、安装python

sudo apt install python3
sudo ln -s /usr/bin/python3 /usr/bin/python
sudo apt install python3-pip

2、安装opencv

sudo apt install python3-opencv

3、安装nvdia显卡驱动

一般来说，如果你安装的是UBUNTU，并在安装时选择了安装第三方显卡驱动程序的话，那么UBUNTU开机就支持驱动了，可以在设置–关于时查看。如果看到如下图图形这时显示为NVIDIA GeForce RTX3060字样，则说明已经安装好驱动了。
在这里插入图片描述
如果你没有安装好驱动或者驱动安装有问题，需要重新安装则可以用以下步骤进行。其中第三个命令会显示当前推荐的驱动，如下图带recommended字样的那个，选择它用第4个命令安装则可。

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo ubuntu-drivers devices
sudo apt install nvidia-driver-525-open

4、安装cuda11.7

参见官方安装连接，cuda11.7是detectron2官方推荐的版本，所以此处安装些版本：

也可以直接参照下面这些命令，依次安装。

wget https://developer.download.nvidia.com/compute/cuda/re

最低0.47元/天解锁文章