This code is based on ScanRefer. Please also refer to the ScanRefer setup.
-
Install PyTorch:
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.2 -c pytorch
-
Install the necessary packages with
requirements.txt:pip install -r requirements.txt
-
Compile the CUDA modules for the PointNet++ backbone:
cd lib/pointnet2 python setup.py install
Note that this code has been tested with Python 3.8, pytorch 1.6.0, and CUDA 10.2 on Ubuntu 18.04.