Windows系统下GroundingDINO的安装2025(踩无数坑)
一、环境配置:
系统:windows10
使用conda的虚拟环境
Python版本:3.8
cuda版本:12.4
torch 版本 : 2.4.0+cu124
torchvision版本:0.19.0+cu124
注:在Windows下编译对环境的要求非常高,版本不对应就会发生各种各样的bug,踩了好多坑发现这个版本可以正常编译,也是挺恶心的
二、环境下载:
(1)创建虚拟环境并进入虚拟环境
conda create -n groundingdino python=3.8
conda activate groundingdino
(2)下载torch和torchvision:
下载地址:https://download.pytorch.org/whl
进入对应页面下载对应版本的whl文件(这样下载比较快,觉得麻烦也可以去pytorch官方找下载指令)
使用命令行进入whl文件所在的文件夹并使用pip install 安装(安装torch和torchvision)
(2)cuda 下载地址:CUDA Toolkit 12.4 Downloads | NVIDIA Developer
下载完之后在系统环境变量里设置CUDA_HOME
二、代码下载:
(1)GroundingDINO代码下载链接:GitHub - IDEA-Research/GroundingDINO: [ECCV 2024] Official implementation of the paper "Grounding DINO: Marrying DINO with Grounded Pre-Training for Open-Set Object Detection"
下载下来之后会如下图所示,看txt文件里面有torch和torchvision(直接pip install -e . 的话装的是CPU版本,如果装的是CPU版本,在真实使用的时候会发生报错NameError: name '_C' is not defined,这是很多人遇到过的问题,所以前面先安装gpu版本的torch)
(2)编译:虚拟环境
命令行进入GroundingDINO所在的文件夹,然后运行下面指令
pip install -e .
不出问题的话就能直接安装成功
三、可能出现的问题
(1)它的编译只支持vs2017到vs2022的,包含2022,如果按照上面运行不能编译成功的话,看看是不是vs版本的问题。
(2)如果vs版本没问题,可以使用进入虚拟环境使用 conda list 看一下虚拟环境的vs,我annaconda默认的是vs2015_runtime,后面我换成vs_2019了(有可能是这个原因导致的,改的东西太多了不确定)