4090显卡安装DCNv2

Linux操作系统上，4090显卡安装DCNv2

安装环境：

Ubuntu24.04，4090

1、首先创建一个虚拟环境，并安装torch

conda create -n JDE python=3.8
conda activate JDE
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch

2、修改cpp_extension.py文件

由于旧版本的pytorch不支持4090显卡，所以需要修改cpp_extension.py文件

文件的位置在miniconda的安装目录/envs/JDE/lib/python3.8/site-packages/torch/utils,其中JDE为环境名称，python3.8为该环境中的Python版本，然后按照下图所示，在这两处添加内容。

 # 在named_arches中添加('Ada', '8.9'),# 在supported_arches中添加'8.9'

3、下载DCNv2的源代码进行编译

# 使用git clone下载，如果下载失败，到github下载（注意选择pytorch_1.7这个分支），然后传到服务器上
git clone -b pytorch_1.7 https://github.com/lbin/DCNv2.git
# DCNv2
https://github.com/lbin/DCNv2

# 若为github下载的需要先进行解压
unzip  DCNv2-master.zip 
# 加入DCNv2目录下
cd  DCNv2-master
# 编译
./make.sh 

安装成功！

在4080服务器上安装若出现找不到torch的情况，使用以下命令进行安装

pip install -e . --use-pep517 --no-build-isolation

安装成功！