背景
spconvが必要になったので、インストールのメモ
テスト環境
- ubuntu18.04
- cuda10.2
- cudnn8.2
- nvidia-driver 440.100
- python3.6.9
- pytorch1.9
インストール
cudnnをdownloadしてdebian packageをinstall
https://developer.nvidia.com/rdp/cudnn-download
sudo dpkg -i libcudnn8_8.2.1.32-1+cuda10.2_amd64.deb
pytorch1.6以上をオススメ(それより前のはバグがありました)
pip3 install -U torch torchvision
spconvをclone
git clone git@github.com:traveller59/spconv.git
boostが必要らしい
sudo apt-get install libboost-all-dev
submoduleのupdate
git submodule init
git submodule update
spconvのinstall
python3 setup.py bdist_wheel
cd ./dist/
pip3 install spconv*.whl
cudnnが見つからない場合
もしcudnnが見つからないエラーが出たら試す
export CUDNN_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu:$CUDNN_LIBRARY_PATH
export CUDNN_INCLUDE_PATH=/usr/include:$CUDNN_INCLUDE_PATHUDE_PATH