pytorch3d 動作確認時メモ

pytorch3dの動作確認です

環境

Ubuntu 18.04 LST
GPU RTX2080Ti
CUDA 10.1
miniconda 4.7.12

環境構築

https://github.com/facebookresearch/pytorch3d/blob/master/INSTALL.md
インストールガイドに沿って環境構築を行いました

git clone

git clone https://github.com/facebookresearch/pytorch3d.git

conda 環境作成

conda create -n pytorch3d python=3.6
conda activate pytorch3d
conda install -c pytorch pytorch torchvision cudatoolkit=10.0
conda install -c conda-forge -c fvcore fvcore

condaでcuda10.0環境を改めて構築する

デモ用環境の構築

pip install scikit-image matplotlib imageio
conda install jupyter
pip install black isort flake8

pytorch3d本体の環境構築

conda install pytorch3d -c pytorch3d

demo実行

notebookを起動させます

cd ./pytorch3d
jupyter-notebook

このnotebook自体では毎回pytorch3dをビルドするのでそこそこ実行時間がかかります
./docs/tutorials内にあるbundle_adjustment.ipynbをnotebookで順次実行していきます

こんな感じで推論されたカメラ角度が出力されるはずです