DN-Detr环境配置

前言

官方代码:

https://github.com/IDEA-opensource/DN-DETR

官方论文:

https://arxiv.org/abs/2203.01305

一、创建一个新的环境:

conda create -n dn_detr python=3.8

激活:
二、激活:

conda activate dn_detr 

三、下载pytorch的各种库:

pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple torch===1.6.0 torchvision===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

这里注意一下,要检测下是否下下载的pytorch版本是不是GPU版本的,必须满足GPU版本,不然使用不了
DN-Detr环境配置_第1张图片

四、然后cd进入主文件
DN-Detr环境配置_第2张图片

安装requirements.txt文件:
官方的requirements.txt文件我改了下,panopticapi和pycocotools 我单独下载了不然下不下来,用我发的这个文件就行了。
改成我这样:
DN-Detr环境配置_第3张图片

五、执行下面三条命令:

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/
pip install panopticapi -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
pip3 install pycocotools -i https://pypi.tuna.tsinghua.edu.cn/simple/ --trusted-host pypi.tuna.tsinghua.edu.cn

六、又进入…\DN-DETR-main\models\dn_dab_deformable_detr\ops文件下面
在这里插入图片描述

python setup.py  build install
python test.py

官方的给定环境条件:
DN-Detr环境配置_第4张图片

你可能感兴趣的:(深度学习,python,pytorch,深度学习)