window11下安装mmrotate0.3.4版本教学

会编码的小板牙

已于 2024-05-13 16:55:54 修改

阅读量892

点赞数 14

文章标签： python pytorch windows

于 2024-03-05 11:46:43 首次发布

本文链接：https://blog.csdn.net/weixin_47658128/article/details/136473936

版权

显卡4060

使用nvidia -smi 查看的驱动版本为12.3

本地服务器的cuda版本为11.8

虚拟环境中pytorch==1.11 cuda==11.3 python==3.8

安装环境代码如下：

conda create -n  mmrotate python=3.8 -y
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch

安装mmrotate之前看看在github上官方给的版本之间的对应关系：MMrotate

我安装的mmrotate版本是0.3.4 所以mmcv-full我选择1.6.0版本，mmdet我安装了2.25.1版本，如下图所示：

pip安装mmcv-full 具有的版本对应可以看这里：mmcv-full

我虚拟环境安装的torch==1.11 cuda==11.3，所以选择对应的mmcv-full如下图，图中代码中的mmcv换成mmcv-full，因为在mmcv2.x版本下的mmcv不包含cuda算子。

pip install -U openmim
pip install mmcv-full==1.6.0 -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.11/index.html

mim install mmdet==2.25.1

安装完pip list查看下自己的包

git clone https://github.com/open-mmlab/mmrotate.git
cd mmrotate
pip install -v -e .

官方给的demo

mim download mmrotate --config oriented_rcnn_r50_fpn_1x_dota_le90 --dest .

运行以下命令

python demo/image_demo.py demo/demo.jpg oriented_rcnn_r50_fpn_1x_dota_le90.py oriented_rcnn_r50_fpn_1x_dota_le90-6d2b2ce0.pth --out-file result.jpg

在mmrotate的目录下看到result.jpg结果

关注