最终结果
终于安装成功了。
环境
系统:ubuntu22.04
CUDA:12.1
python:3.11
显卡驱动:545
安装过程
系统、CUDA和python的安装过程忽略,这些都能找到。这里只写不同的地方。
配置CUDAHOME
执行命令“
gedit ~/.bashrc
加入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
然后,执行命令:
source ~/.bashrc
解决权限不够的问题
如下:
<command-line>: fatal error: /usr/local/cuda-12.1/include/stdc-predef.h: 权限不够
compilation terminated.
解决方法
进入到cuda-12.1文件夹下面,启动终端,如下图:
修改include权限,执行命令:
sudo chmod 777 include
然后,再接着编译
按装ninja
编译的时候用到了ninja,安装方法:
pip install ninja
编译DCNv4
进入到DCNv4_op文件夹下面,如下图:
然后,执行编译命令:
python setup.py build install
最终就可以完成编译了!