ubuntu20 使用conda安装mmdetection环境

一、创建虚拟环境
conda create -n open-mmlab python=3.7 -y
conda activate open-mmlab

二、安装pytoch + cudatoolkit
去pytorch官网 https://pytorch.org/ 按照自己的要求来选择,根据官网的命令来就好:

这里建议不要安装cuda11,因为后面直接使用apt命令安装的nvidia驱动是针对cuda10版本的,与cuda11不匹配,这是因为apt命令安装的驱动不是最新的,想要安装最新的与cuda11匹配的驱动,应该需要自己去官网上扒一下。
所以这里建议使用cuda10对应的命令:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

三、安装nvidia驱动
在命令行使用
nvcc -V
命令看一下驱动版本,如果显示找不到,就说明还没有安装驱动,使用下面命令安装驱动:
sudo apt install nvidia-cuda-toolkit

四、安装mvcc-full
这一步可以参考这篇博客:https://blog.csdn.net/qq_44442727/article/details/113444207
去https://github.com/open-mmlab/mmcv上面,根据自己的cuda版本和torch版本,找到对应的安装命令,安装即可

五、安装mmdetection
此步骤参考博客:https://blog.csdn.net/qq_44442727/article/details/113444207

六、验证
安装完成后,输入命令
python mmdet/utils/collect_env.py

会输出一系列信息,注意里面cuda available应该是true,则证明安装成功

posted @   彼岸的客人  阅读(382)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· AI编程工具终极对决:字节Trae VS Cursor,谁才是开发者新宠?
· 开源Multi-agent AI智能体框架aevatar.ai,欢迎大家贡献代码
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
点击右上角即可分享
微信分享提示