how to setup pointmamba - n-eisei/PointMamba GitHub Wiki

ステップ

前提：ホストマシンにCUDA nvidiaドライバとcondaがインストール済。

$ conda create -n pointmamba python=3.9 cudatoolkit=11.7
$ conda activate pointmamba
$ ./install.cmd
学習データをダウンロード
- 下記URLに参照しながら、プロジェクトフォルダにdataを準備する
- データ準備はここ参照：　https://github.com/n-eisei/PointMamba/blob/main/DATASET.md
$ CUDA_VISIBLE_DEVICES=0 python main.py --config cfgs/pretrain.yaml --exp_name mypj_exp_name

※　注意：ステップ3,5を実行する前に、"conda activate pointmamba"コマンドでpython開発環境に入っていること。

install.cmdの内容

pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117

# Install basic required packages
pip install -r requirements.txt

# Chamfer Distance & emd
cd ./extensions/chamfer_dist && python setup.py install --user
cd ./extensions/emd && python setup.py install --user

# PointNet++
pip install "git+https://github.com/erikwijmans/Pointnet2_PyTorch.git#egg=pointnet2_ops&subdirectory=pointnet2_ops_lib"

# GPU kNN
pip install --upgrade https://github.com/unlimblue/KNN_CUDA/releases/download/0.2/KNN_CUDA-0.2-py3-none-any.whl

# Mamba
pip install causal-conv1d==1.1.1
pip install mamba-ssm==1.1.1