本来我之前都是在YOLO上跑的,已经搭建好了虚拟环境,所以我本来计划是在这个环境上直接部署,来跑RT-DETR模型。但是遇到了一系列问题,所以写下这篇文章,希望和我遇到同样情况的宝子们可以避免出错!!!
首先要给大家说的是,跑RT-DETR时要重新创一个虚拟环境,不要为了省事在原来的环境上安装库,这样后期会出现很多问题。
先打开Anaconda Prompt窗口 输入
conda env list
这样就会显示出你目前有的虚拟环境
这是我所创建的虚拟环境(有点多,因为前期总是因为版本问题和兼容性问题出错,所以创建了比较多的环境),里面的yolov10(这是我起的虚拟环境名称,不是针对某个程序,这点小伙伴们不要弄混)环境出错,CUDA不能正常使用,经过我的查询,发现问题应该是我的python版本(3.12.4)与Pytorch版本不兼容,所以在这里也建议大家安装python时不要安装最新版本的(我建议安装3.8或者3.9,大多数模型都能兼容)。
然后我删除yolov10这个虚拟环境
输入命令 conda env remove --name yolov10
然后再输入 conda env list 检查是否删除了指定的虚拟环境
从上图可以看出来已成功删除。
接下来就是创建RT-DETR所需的虚拟环境(具体的版本我希望大家按照我安装的版本来,这样避免出现不兼容情况),我把虚拟环境名称定义为rtdetr(是因为环境太多,害怕后面自己记混,在这里也建议大家自己创建虚拟环境的时候也起一个相关的名称)</