此帖記錄一下SOLO環境配置的過程以及一些問題和解決辦法。

步驟

1.通過git下載SOLO源碼

git clone https://github.com/WXinlong/SOLO.git cd SOLO

在使用git命令時，anaconda prompt不能識別，但是cmd可以識別，對此直接conda install git（這一步真是費了好多時間）

2.安裝依賴

pip install -r requirements/build.txt

3.安裝pycocoapi

pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI

不能用下面這種，會出錯

pip install "git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI"

4.配置mmdet，最好和官方文檔要求一致

pip install mmcv==0.2.16

5.cuda版本過高不行，比如我第一次時cuda11.3，后來換了cuda10.1，RTX30系列顯卡好像不支持10.x，唉TAT。安裝pytorch，這里參考了這篇文章cuda10.1+pytorch1.7配置centernet環境
配置清華鏡像conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --set show_channel_urls yes conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
安裝pytorch

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1

6.編譯

python setup.py develop

7.環境測試（這一步參考【SOLO】環境配置（mmdetection=1.0.0+mmcv = 0.2.15））
SOLO目錄下創建checkpoints文件夾，存入model文件，Decoupled_SOLO_R50_3x：

md checkpoints cd checkpoints copy C:\Users\admin\Downloads\DECOUPLED_SOLO_R50_3x.pth cd.. cd demo python inference_demo.py demo_out.jpg

最后結果如圖所示：

總結

以上是生活随笔為你收集整理的【solo】环境配置的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。