mit-han-lab/pvcnn

can not import cpp files?

coallar opened this issue · 1 comments

coallar commented

Hi, it is wrong when i train the command:
python3 train.py config/shapenet/pvcnn/c1.py
details coud be senn as below:

image

zhijian-liu commented

This is probably due to the CUDA environment configuration. You could add these lines into your .bashrc:

export CUDA_PATH=/usr/local/cuda
export PATH=$CUDA_PATH/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_PATH/lib64:$LD_LIBRARY_PATH

Remember to restart the terminal after this.