Windows 10 の WSL2 上 の Ubuntu 20.04 で Detectron2 をインストール
Detectron2 をインストールするのは容易で、公式HPにある、
python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'
で終わり。
と言いたいのだが、2022年5月現在、いろいろ問題がおきまくってぜんぜんうまくいかなかったので解決方法を記述する。
ninjaがいないと怒られる
Ninjaが必要だと。
sudo apt install -y ninja-build
wheelがいないと怒られる
wheel無いって。
pip install wheel
pycocotools でコンパイルエラー
command: /usr/bin/python3 -u -c 'import sys, setuptools, tokenize; sys.argv[0] = '"'"'/tmp/pip-install-bnn8kw21/pycocotools_f13f62aaf49546019dc9923edf4326fe/setup.py'"'"'; file='"'"'/tmp/pip-install-bnn8kw21/pycocotools_f13f62aaf49546019dc9923edf4326fe/setup.py'"'"';f=getattr(tokenize, '"'"'open'"'"', open)(file);code=f.read().replace('"'"'\r\n'"'"', '"'"'\n'"'"');f.close();exec(compile(code, file, '"'"'exec'"'"'))' bdist_wheel -d /tmp/pip-wheel-zbqqcdat
エラーになるのがおかしいのだけど、みんな、別途インストールすることで回避してるから別途インストール。
pip install pycocotools
pipが古いせいでコンパイルエラー
さらになんかエラーになるけど、pipが古いと起きるんだって。
Building wheel for detectron2 (setup.py) ... error ERROR: Command errored out with exit status 1:
ので、バージョンを最新のを指定してインストール。
pip install pip==22.1.2
CUDAのバージョンおかしいって！
2022年5月現在、Nvidiaの公式ページに従ってWSL用のCUDAを入れると11.7になる。
が、
RuntimeError:
The detected CUDA version (11.7) mismatches the version that was used to compile
PyTorch (11.3). Please make sure to use the same CUDA versions.
と、11.3をご所望のよう。
なので、11.3を入れる。
wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-wsl-ubuntu.pin
sudo mv cuda-wsl-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda-repo-wsl-ubuntu-11-3-local_11.3.1-1_amd64.deb
sudo dpkg -i cuda-repo-wsl-ubuntu-11-3-local_11.3.1-1_amd64.deb
sudo apt-key add /var/cuda-repo-wsl-ubuntu-11-3-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda-repo-ubuntu2004-11-3-local_11.3.1-465.19.01-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-3-local_11.3.1-465.19.01-1_amd64.deb
wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-keyring_1.0-1_all.deb
sudo dpkg -i cuda-keyring_1.0-1_all.deb
sudo apt-get update
sudo apt-get -y install cuda-toolkit-11-3
export CUDA_PATH=/usr/local/cuda-11.3
echo 'export CUDA_PATH=/usr/local/cuda-11.3' >> ${HOME}/.bashrc
export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64:${LD_LIBRARY_PATH}
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64:${LD_LIBRARY_PATH}' >> ${HOME}/.bashrc
export PATH=/usr/local/cuda-11.3/bin:${PATH}
echo 'export PATH=/usr/local/cuda-11.3/bin:${PATH}' >> ${HOME}/.bashrc
source ~/.bashrc
ここまでやって通らなかったら知らん。
あ、python-devないとだめかも。
sudo apt install python3-dev
Detectron2 や D2Go で Segmentation fault
Detectron2 を importしてpython実行すると Segmentation fault が。
これ、ちゃんとコンパイルせずに、以下の公式HPのInstall Pre-Built Detectron2 (Linux only) のをインストールしたせい。
https://detectron2.readthedocs.io/en/latest/tutorials/install.html
これは普通のLinux用のやつでWindows上のUbuntuのバイナリじゃないから動かしたときに Segmentation fault が発生してしまった。
PyTorchはPythonなので油断しがちだけどちょいちょいCのバイナリでてくるから。
WSL上のUbuntuやMacで Detectron2 や D2Go を使いたいならコンパイルすること。
以上！