【CV】MMDetction(MMCV)安装踩坑

踩雷历程(可直接看经验总结)

先是没看清mmdetection官网中说的pytorch版本要求,导致以下报错

ERROR: Could not build wheels for mmcv, which is required to install pyproject.toml-based projects

再是不知道如何选择CUDA和pytorch的版本,就根据自己的电脑显卡随便安装了一个可行的CUDA(10.2),然后又根据pytorch官网中选择了本版本的CUDA可安装的torch,然后报错,这应该是pytorch版本不够。

NVIDIA GeForce RTX 3060 Laptop GPU with CUDA capability sm_86 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.
If you want to use the NVIDIA GeForce RTX 3060 Laptop GPU GPU with PyTorch, please check the instructions at 

可是现在升级pytorch又因为依赖项问题不成功,十分崩溃。所以总结了一下安装MMDetction的方法(主要参考了这个博主的思路

经验总结:根据MMCV选择合适的CUDA和pytorch

这个博主的思路提到了用以下方法安装:

mim install mmcv==2.0.0 -f https://download.openmmlab.com/mmcv/dist/cu117/torch1.13/index.html

于是在确定版本的时候,便可以先选择硬件允许的尽量高的CUDA版本,然后去pytorch官网查找可安装的尽可能高的pytorch版本,之后再把这两个版本填入以下连接

https://download.openmmlab.com/mmcv/dist/cu版本/torch版本/index.html

然后输入到浏览器地址栏查找时候有对应的包可以下载,有则表示CUDA版本和Pytorch版本OK,没有则换一个CUDA版本和Pytorch版本,知道找到一个有包可以下载的情况,然后即可用当前的CUDA版本和Pytorch版本。

无包(重新试一下其他版本):

有包(就用这个CUDA版本和Pytorch版本):

  • 9
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值