说一下踩得坑:原来不懂,上去就pip install,结果安装上却不能用,因为版本不对

------------------------------------------------------------------------------------------------------------------

方法:

首先要安装mmcv要去官网查看一下对应版本:GitHub - open-mmlab/mmcv: OpenMMLab Computer Vision Foundation

 目前官网最高只支持CUDA11.3版本,没事的,也是兼容CUDA11.5的。

打开上面那个网站,看一下都支持什么版本的mmcv。 

我安装的是最新的1.4.2

pip install mmcv-full==1.4.2 -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.10.0/index.html

 

 

 

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐