复现 Oriented R-CNN RTX3080ti

在3080ti主机上复现Oriented R-CNN，参考了：复现Oriented R-CNN RTX 2080Ti_吃肉不能购的博客-CSDN博客

这篇博文给我提供了很大帮助

前言

GitHub链接：GitHub - jbwang1997/OBBDetection: OBBDetection is an oriented object detection library, which is based on MMdetection.

显卡：RTX3080ti

系统：Ubuntu20.04 

一、创建环境

conda create -n pytorch171 python=3.7 -y
conda activate pytorch171
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch

这里我一开始用了我的比较高版本的pytorch去跑，发现会报THC/THC.h的错误，简单说就是torch版本太高，和代码里用的函数不匹配。之后参考另一篇博文想用pytorch=1.6.0，但是30系显卡只支持cuda11，所以尝试了最低的支持cuda11的版本torch=1.7.0，然后代码setup成功了，但还是莫名其妙有错误，查了说好像需要1.7.1。第二天索性删掉环境重来，直接下载torch1.7.1。 

二、配置OBBdetection

下载代码文件

git clone https://github.com/jbwang1997/OBBDetection.git --recursive
cd OBBDetection

安装BboxToolkit

cd BboxToolkit
pip install -v -e .  
cd ..

 如果下载git下载的时候出错误了BboxToolkit没下载下来，或者文件夹没有setup.py文件，就去https://github.com/jbwan