MMdetection trainの環境構築

mmdetectionでtrainしようとしたが、どハマりしたので備忘録

公式ドキュメントの通りにやるとハマった。

環境　
ubuntu20.04
RTX3060 Laptop
CUDA 11.6 ←ここ大事　あとからインストールするMMCVのバージョンに関わる

conda create --name openmmlab python=3.8 -y
conda activate openmmlab 
#仮想環境を作る

conda install pytorch==1.12.0 torchvision -c pytorch 
#pytorchを仮想環境にインストール 1.12が大事

pip install -U openmim 
#mmlabのパッケージ管理モジュールmimを入れる

mim install mmcv-full==1.6.0 
#バージョン指定しないと最新版1.7.1が入る

mmdetの中のなんちゃらpyファイルがバージョンを指定していてそれが1.6.0までのはず…詳しくは忘れたfaq.mdにも書いてある
1.7.0だとtrain.pyの20行〜22行をimportできなかった
https://mmcv.readthedocs.io/en/latest/get_started/installation.html#install-mmcv-full ここで1.6.0を入れられる条件確認

git clone https://github.com/open-mmlab/mmdetection.git

cd mmdetection

git log

#2.25.0のハッシュ値をコピー

git checkout "ハッシュ値"

# mmdetectionのバージョンを2.25.0に切り替える

https://qiita.com/sister_DB/items/3d931f713681d8ba5c4a こんな記事を見つけたのでそのままやるのが嫌だから

pip install -v -e .

python
#インタラクティブモード
import torch
torch.cuda.is_available()

true #環境構築完了

false #torchのホイールをpip install