detectron2安装踩坑记录Win+Ubuntu
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版本高了没啥大问题,编译是能过去的。要是你报了相关错我估计是依赖项没有安装合适