ディープラーニングを用いた3次元点群レジストレーションを動かす
下記を動かす。
https://github.com/chrischoy/FCGF
環境構築
ディープラーニング系ライブラリのDockerイメージをまとめたDeepoを利用する。
FCGF要求のpython3.7 + pytorchな環境はビルド済みタグにはない。
ので自作する。
FCGFダウンロード
cd ~/Documents
git clone https://github.com/chrischoy/FCGF
Dockerイメージのベース作成
cd ~/Documents
git clone https://github.com/ufoym/deepo.git
cd deepo/generator
ython3 generate.py --cuda-ver 10.1 --cudnn-ver 7 Dockerfile pytorch python==3.7
docker build -t my/deepo .
※cudaやUbuntuのバージョンによっては、組み合わせが作れないことがある。
下記のdockerhubからベースイメージを得ているので、
下記で存在するものを指定する。
https://hub.docker.com/r/nvidia/cuda/tags?page=1&name=ubuntu
ベースイメージに必要パッケージをインストール
docker run --gpus all -it --rm -v $HOME/Documents:/Documents my/deepo bash
apt update
apt-get install libblas-dev
apt-get install liblapack-dev
apt-get install gfortran
cd /Documents/FCGF
pip install -r requirements.txt
apt-get install -y libgl1-mesa-dev
※ Minkowski Engineのインストールにはかなり時間がかかる。
Dockerイメージを更新
docker commit {起動中のコンテナID} my/deepo:ver2