SPVNAS Installation

Installation guide for SPVNAS repository

(code): https://github.com/mit-han-lab/spvnas

1. Conda 환경 생성

conda create -n spvnas python=3.7
conda activate spvnas

2. Cuda 11.3 설정

conda install cudatoolkit=11.3 -c conda-forge -y
conda install nvidia-/label/cuda-11.3.1::cuda-toolkit

3. Pytorch 및 필요 라이브러리 설치

conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge
conda install numba=0.53 opencv=4.2.0 -c conda-forge
pip install torchpack

4. (까다로운) torchsparse 설치

• 이 부분에서 계속 에러가 발생하였다. -> 크게 3가지 문제: Ninja가 conda에 다운되어야 하며, google-sparsehash를 추가 설치, 마지막으로 gcc버전이 10 미만이어야 한다.

1. Ninja 설치

conda install -c conda-forge ninja

2. Sparsehash 설치

conda install bioconda::google-sparsehash

3. gxx 9.3 버전 설치

conda install gxx_linux-64=9.3.0
export CC=$(which x86_64-conda-linux-gnu-gcc)
export CXX=$(which x86_64-conda-linux-gnu-g++)

이 후 아래 command를 통해 설치 진행

pip install --upgrade git+https://github.com/mit-han-lab/torchsparse.git