1.创建一个conda env
conda create -n MMRotate python=3.8
2.激活环境
conda activate MMRotate
3.安装pytorch(对应cuda版本,我的是10.1)
pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
4.测试pytorch
import torch
torch.cuda.is_available()
5.安装mmcv-full(找好对应版本)
https://mmcv.readthedocs.io/en/latest/get_started/installation.html
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.8.0/index.html
pip下的慢,更换一下pip的源
在主目录下创建.pip文件夹,然后在该目录下创建pip.conf文件
mkdir ~/.pip
vim ~/.pip/pip.conf
更换为清华大学镜像源
[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
6.安装mmdet
pip install mmdet
7.安装mmrotate
pip install mmrotate
conda list 查看都已安装好
或者
git clone https://github.com/open-mmlab/mmrotate.git
cd mmrotate
pip install -r requirements/build.txt
pip install -v -e . # or "python setup.py develop"
要用源码早晚都得git下来