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,则证明安装成功
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· AI编程工具终极对决:字节Trae VS Cursor,谁才是开发者新宠?
· 开源Multi-agent AI智能体框架aevatar.ai,欢迎大家贡献代码
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!