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了(有可能是这个原因导致的,改的东西太多了不确定

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值