ubuntu编译dcnv3问题_notimplementederror: cuda is not availabel

Dcnv3源码

源码中选择要进行的视觉任务下的ops_dcnv3拷贝到工作区编译源码

$cd ops_dcnv3
$sh make.sh

NotImplementedError: Cuda is not availabel

普通用户需要找到管理员用户下安装的cuda，默认路径在/home/usr/local

$export CUDA_HOME=$CUDA_HOME:/home/usr/local/cuda-x.x
$export PATH=$PATH:/home/usr/local/cuda-x.x/bin

dcnv3_im2col_cuda.cuh:17:10: fatal error: ATen/OpMathType.h: 没有那个文件或目录

$export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/usr/local/cuda-x.x/lib64

AttributeError: ‘NoneType‘ object has no attribute ‘_free_weak_ref‘

 设置训练参数的进程workers=0

DCNv3-1.0配置

$source /home/usr/anaconda/bin/activate
$pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113 -i https://pypi.tuna.tsinghua.edu.cn/simple