- 博客(4)
- 收藏
- 关注
RSS订阅原创 “from mmcv.ops import ModulatedDeformConv2d“标红或者mmcv找不到的问题
大部分人装mmcv时，如果直接pip install，那么可能遇到的问题是mmcv和pytorch版本不对应。可以看到截止目前（2025.3.4日），torch版本只支持2.1.x版本的，cuda最高只支持12.1的。已经配置好清华等镜像源的伙伴可以直接去除命令的红框部分，默认从自己的源下载，速度快。否则参考刚刚第一条的。，则说明安装了cuda且版本是12.1的，且pytorch版本为2.1.0（GPU版本）。如果GPU版本的默认下载方式太慢，因为从国外下载，那么可以参考这个文章，从阿里云下载。
2025-03-04 00:58:25
535
3
原创 torch、torchvision、torchaudio安装对应cuda版本（cuda 12.1情况下）
如果安装cuda 11.8版本的，把最后的cu121改成cu118即可。目前支持最高cuda 12.2。因为网上 pip install torch 等全部默认安装cpu版本，推理速度慢。所有安装cuda版本的包会更快。
2024-07-03 19:44:41
10102
2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人