Pytorch环境安装
显卡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版本,如下图所示:
安装mmcv-full
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
安装mmdet
mim install mmdet==2.25.1
安装完pip list查看下自己的包
安装mmrotate
git clone https://github.com/open-mmlab/mmrotate.git
cd mmrotate
pip install -v -e .
验证mmrotate
官方给的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结果