安装DCNv2 记录

本文讲述了在安装DCNv2时遇到的常见问题,包括torch版本过高或过低导致的编译错误,以及torch与cuda版本不匹配的情况。作者分享了解决方案,如调整torch版本、确保cuda版本对应,以及注意conda和pip混合安装可能导致的冲突。
摘要由CSDN通过智能技术生成

1. 普遍方法(可以解决大多数问题)

首次安装出现的问题

类似下面这样

raise RuntimeError(message) from e RuntimeError: Error compiling objects for extension

由于需要较新的torch库,所以没办法降低太多torch版本

可能原因及方法

1) torch版本太高(>torch 2.0.1)

可以降低torch版本,同时保证降低版本后cuda版本也匹配;

列举一些DCNv2的项目库:

原始项目:https://github.com/CharlesShang/DCNv2

(只有低版本的pytorch安装)

推荐 :https://github.com/lbin/DCNv2

有较多版本供选择:
在这里插入图片描述
我在torch 2.0.1版本上可以用,算是比较新了

也是一个库: https://github.com/lucasjinreal/DCNv2_latest

有博主不推荐,我懒得试了

注:没试过在太高版本下的安装(>torch 2.0.1),大家也可以用相近版本的项目库的代码在高版本torch上尝试。

2)torch版本不高(<=torch 2.0.1),但安装不成功

可能torch和cuda版本不对应,上网检查一下

import torch
print(torch.__version__)  
print(torch.cuda.is_available())  # cuda是否可用
print(torch.version.cuda)  # cuda版本

举例:
在这里插入图片描述
在这里插入图片描述

3)前面的都对,但还不成功

可能和torch有关的库是用conda和pip混合安装的,导致出错
用命令检查一下库的安装,卸载统一用pip

conda list

2. 我遇到的问题

上面遇到的都对,但就是不行。后面在多个环境下尝试,反复拉扯,终于可以了。
发现我有这么两个包,这两个是用conda安装的:
cudatoolkit
cuda-nvcc

于是我在一个本来可以安装DCNv2的环境安装这两个包后,DCNv2就不能再安装了。于是我卸载了这两个包,

conda remove cudatoolkit
conda remove cuda-nvcc

终于在 pytorch2.0.1+cuda118 成功。

  • 6
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
要在Windows上安装dcnv2,您需要按照以下步骤进行操作: 1. 首先,您需要安装NVIDIA GPU驱动程序。请确保您的计算机上已安装最新版本的NVIDIA GPU驱动程序。 2. 接下来,您需要安装CUDA Toolkit。dcnv2需要CUDA支持才能正常工作。您可以从NVIDIA官方网站下载并安装适用于您的GPU的CUDA Toolkit。 3. 安装cuDNN库。cuDNN是一个GPU加速库,用于深度神经网络。您可以从NVIDIA开发者网站下载适用于您的CUDA版本的cuDNN库,并按照安装说明进行安装。 4. 下载并安装Anaconda。Anaconda是一个常用的Python发行版,它包含了许多科学计算和深度学习所需的包和工具。 5. 创建并激活一个虚拟环境。打开Anaconda Prompt(在开始菜单中搜索Anaconda Prompt),然后使用以下命令创建一个名为dcnv2的虚拟环境: ```shell conda create -n dcnv2 python=3.8 ``` 然后激活该虚拟环境: ```shell conda activate dcnv2 ``` 6. 安装PyTorch和TorchVision。在激活的虚拟环境中运行以下命令: ```shell conda install pytorch torchvision cudatoolkit=xx.x -c pytorch ``` 注意,将"xx.x"替换为您安装的CUDA版本对应的数字。例如,如果您安装了CUDA 11.0,则应使用"cudatoolkit=11.0"。 7. 安装其他依赖项。在虚拟环境中运行以下命令安装其他必需的Python包: ```shell pip install opencv-python pillow matplotlib ``` 8. 下载并安装dcnv2。您可以从dcnv2的GitHub存储库中下载源代码,并按照其中的说明进行安装和配置。 完成以上步骤后,您应该能够成功在Windows上安装dcnv2。请确保按照上述步骤中的说明进行操作,并根据具体情况调整相关参数和版本号。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值