本来我之前都是在YOLO上跑的,已经搭建好了虚拟环境,所以我本来计划是在这个环境上直接部署,来跑RT-DETR模型。但是遇到了一系列问题,所以写下这篇文章,希望和我遇到同样情况的宝子们可以避免出错!!!
首先要给大家说的是,跑RT-DETR时要重新创一个虚拟环境,不要为了省事在原来的环境上安装库,这样后期会出现很多问题。
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(是因为环境太多,害怕后面自己记混,在这里也建议大家自己创建虚拟环境的时候也起一个相关的名称)
输入命令 conda create -n rtdetr python=3.9

输入 y 表示确定安装
之后会提示安装成功

然后进入自己刚创建的虚拟环境
输入命令 conda activate rtdetr
观察到前面的(base)变成了(rtdetr)
然后输入 python
会弹出你刚安装好的虚拟环境里面的python信息
再输入 exit() 退出到虚拟环境中安装torch、cuda等

下面安装torch、cuda的命令要和我之前文章里提到的那样,去pytorch官网找具体的命令行(如果大家忘记可以翻一下我之前发的文章,里面有链接),这里我直接附上我安装的具体命令
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
会提示安装成功

下面要注意,我本来以为到这里就完成了对torch、cuda的安装,但是当我进行验证时却报错

经过查询才知道,是此时默认安装的Numpy版本不匹配,默认安装的是2.0.2,我查阅资料告诉我要用1.19.x - 1.21.x。于是我把原来的Numpy卸载,安装了1.19版本的,但还是报错,如下:


经过我几次的尝试,我发现需要安装的Numpy版本是1.20.3

下面就是验证刚才安装的torch、cuda等是否安装成功
首先输入 python
再输入
import torch
import torchvision
print(torch.__version__)
print(torchvision.__version__)
print(torch.cuda.is_available())
如果你按照我上面的步骤,应该就不会出错,成功显示的页面如下:

再然后,就需要安装RT-DETR所需的timm库,为了避免兼容性问题,请大家安装0.5.4这个版本,我已经试过,成功安装
输入命令 pip install timm==0.5.4

接着 输入命令
pip install -U openmim
mim install mmengine
mim install mmcv==2.0.1 mmcv最好也安装这个版本,避免出错
最后输入 pip list 查看自己虚拟环境里面刚安装的库

到这里基本就结束了。
但是请大家注意,上面这是把只要的几个库给安装成功了,在实际运行代码时候,还会有需要安装好的库,到时候按需安装。 输入命令 pip install xxx(库的名称)
还有一点我要强调一下,因为现在很多RT-DETR的代码都是在ultralytics集成框架上进行编写的,所以大家在运行RT-DETR时候,不要安装ultralytics,这是集成框架里面的一个文件夹,不是需要安装的库,这点需要注意!!!
感谢您的观看,如果有用,麻烦动一下您的指头给颗小红心
您的支持是我更新的动力。
到此这篇conda虚拟环境列表(conda建虚拟环境)的文章就介绍到这了,更多相关内容请继续浏览下面的相关推荐文章,希望大家都能在编程的领域有一番成就!版权声明:
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如若内容造成侵权、违法违规、事实不符,请将相关资料发送至xkadmin@xkablog.com进行投诉反馈,一经查实,立即处理!
转载请注明出处,原文链接:https://www.xkablog.com/cjjbc/16663.html