CUDA10.1 + python3.7 + pytorch1.7 安装DCNv2

最近想运行一个网络,用到了centernet,需要安装DCNv2,折腾了两天,记录一下填坑的过程。

  1. 下载
    (1)github上最新版的DCNv2源码在"https://github.com/CharlesShang/DCNv2",但是该版本源码不支持PyTorch1.7,如果使其支持PyTorch1.7需要做以下修改:
    DCNv2/src/cuda/dcn_v2_cuda.cu中所有的THCState_getCurrentStream(state)替换成c10::cuda::getCurrentCUDAStream()

(2)github上有支持PyTorch1.7的源码,“https://github.com/lbin/DCNv2/tree/pytorch_1.7”,可以直接使用。

  1. 修改torch/utils/cpp_extension.py文件
    该文件一般在你的python安装目录site-packages文件夹下,比如我的在conda虚拟环境路径/lib/python3.7/site-packages/torch/utils/cpp_extension.py
    (1)将源码command = [‘ninja’, ‘-v’]修改成command = [‘ninja’, ‘–version’],注意version前面是两条短横线。
    (2)将源码compiler_info.decode().strip()修改成compiler_info.decode(‘gbk’).strip()

  2. 编译
    DCNv2目录下执行./make.sh即可。

  3. 错误
    在使用conda

评论 11
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值