【detectron2安装踩坑记录】

Windows11

前期准备

为了防止出现高版本CUDA跟低版本编译不兼容的问题（比如说我装了12.0的CUDA，然后torch用cuda11.8，这样普通安装pytorch没问题，但是在采用detectron2编译的过程中，它有可能会给警告或者报错什么的，我就索性直接把电脑的cuda版本改成12.1，这样就跟最新的torch2.2.1要求的一样了）

我发现这个detectron2安装过程种torch版本可以是latest，但是python版本最好不能太高，不然容易寄了，反正3.8能安装成功，大于3.10还是建议谨慎一点。

我安装如下版本的torch

切换到一个比较空的目录下，也就是你想存放detectron2项目本身的敌方

安装依赖项

安装cython，oprncv-python，pillow，matplotlib，tercolor。。。

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
pip install cython opencv-python pillow  matplotlib termcolor cloudpickle tabulate tensorboard termcolor tqdm yacs mock fvcore pydot wheel future

pip install pycocotools

pip install pywin32

pip install ninja

安装detectron

git clone https://github.com/facebookresearch/detectron2.git
cd detectron2

pip install -e .

验证安装

把notebook也装了然后开一个notebook，贴入以下代码（detectron2有一个官方的colab指导安装，这段代码是我从官方的colab抄来的）

import torch, detectron2
!nvcc --version
TORCH_VERSION = ".".join(torch.__version__.split(".")[:2])
CUDA_VERSION = torch.__version__.split("+")[-1]
print("torch: ", TORCH_VERSION, "; cuda: ", CUDA_VERSION)
print("detectron2:", detectron2.__version__)

这就说明安装成功了
colab地址：

https://colab.research.google.com/drive/16jcaJoc6bCFAQ96jDe2HwtXj7BMD_-m5#scrollTo=FsePPpwZSmqt

Ubuntu2004

previous build版本(LINUX ONLY)

这个我直接安装的CUDA1.10加torch1.10版本的previous build版本
这个官网有教程告诉你咋弄，但是就是他的python版本torch版本和cuda版本都得安装特定的，而且2021年以后就再也没有更新过了，还是推荐下载源码自己编译的方法（当然这个就不用本地编译了）

本地编译

要是按照上面的windows下的安装过程在本地编译也基本没问题，唯一不同的是在ubuntu下pywin32这个库是装不上的，但是这个报错无关紧要，好像这个库也不是ubuntu下detectron2所必需的，我这次尝试了python3.10发现是没问题的，看下边的中终端里面，不要管那个nvcc啥的，反正torch和detectron2都装成功了
补充：我ubuntu下nvidia-smi出来的CUDA版本是12.3，看样子torch版本高了没啥大问题，编译是能过去的。要是你报了相关错我估计是依赖项没有安装合适