Linux配置mmcv-full

Linux配置mmcv-full,第1张

1、根据当前cuda版本安装对应torch版本,本人cuda10.2 ,torch安装的是1.7.12、结合mmdetection的版本,如下图,因为我安装的是最新版的mmdetection,已经到了2.8.0版本了,mmcv-full的版本要求在>=1.2.4,<1.3于是我选择1.2.4版本。

 3、安装mmcv-full,以下命令中的  https://download.openmmlab.com/mmcv/dist/cu102/torch1.7.0/index.html需结合自己当前cuda以及torch版本来进行修改本人安装的cuda102,torch1.7.1。

pip install mmcv-full==1.2.4 -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.7.0/index.html

4、完成!

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/736760.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-04-28
下一篇 2022-04-28

发表评论

登录后才能评论

评论列表(0条)

保存