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、完成!
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)