背景:
显卡4060
使用nvidia -smi 查看的驱动版本为12.3
本地服务器的cuda版本为11.8
虚拟环境中pytorch==2.1.0 cuda==11.8 python==3.8
pytorch安装命令
conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=11.8 -c pytorch -c nvidia
如下图mmcv选择自己对应的torch和cuda版本安装
mmdet我是使用mim命令
mim install mmdet
原因分析:
查了相关资料,可能是版本不兼容问题。但我torch cuda python mmcv mmdet之间版本兼容是对的。
之后又试了试mmcv 和 mmdet源码安装,不出所料还是报错。
然后听说要安装mmcv-full,把mmcv删除掉。
在虚拟环境中把mmcv删除,使用mim命令安装mmcv-full。
mim install mmcv-full
运行官方demo还是会报错误。
解决方案:
后面我删除掉虚拟环境中的mmdet
重新mim install mmdet
由于安装的是当前最新版的mmdet==3.3.0,需要mmcv2.1.0来兼容,mmcv又下回来了。
此时我的pip list查看的包是这样的
很奇怪的是mmcv和mmcv-full一起下了反而不会报错,单独下了mmcv2.1.0和mmdet3.3.0不行,加上一个mmcv-full就可以,运行demo代码也没有报错。