成功安装DCNv4_pip install dcnv4

最终结果

终于安装成功了。

环境

系统：ubuntu22.04
CUDA：12.1
python：3.11
显卡驱动：545

安装过程

系统、CUDA和python的安装过程忽略，这些都能找到。这里只写不同的地方。

配置CUDAHOME

执行命令“

gedit ~/.bashrc
1

加入CUDAHome，如下图：

export PATH=/usr/local/cuda-12.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

export CUDA_HOME=/usr/local/cuda-12.1
1
2
3
4

然后，执行命令：

source ~/.bashrc

1
2

解决权限不够的问题

如下：

<command-line>: fatal error: /usr/local/cuda-12.1/include/stdc-predef.h: 权限不够
compilation terminated.

1
2
3

解决方法
进入到cuda-12.1文件夹下面，启动终端，如下图：

修改include权限，执行命令：

sudo chmod 777 include
1

然后，再接着编译

按装ninja

编译的时候用到了ninja，安装方法：

pip install ninja
1

编译DCNv4

进入到DCNv4_op文件夹下面，如下图：

然后，执行编译命令：

python setup.py build install
1

最终就可以完成编译了！