HKUST-Aerial-Robotics / Stereo-RCNN

Code for 'Stereo R-CNN based 3D Object Detection for Autonomous Driving' (CVPR 2019)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

RuntimeError: Not compiled with GPU support Stereo-RCNN/lib/model/csrc/nms.h:22

haleqiu opened this issue · comments

When running python3 demo.py I met a problem running nvidia-docker
with cuda 9.0 torch 1.01 torchvision 0.4.0

After python3 setup.py build develop
RuntimeError: Not compiled with GPU support (nms at /home/yuhengq/Stereo-RCNN/lib/model/csrc/nms.h:22)

I met a problem running nvidia-docker with cuda 9.0 torch 1.01 torchvision 0.4.0

After python3 setup.py build develop, I run python3 demo.py which return the error
line 121, in forward
keep_idx_i_left = nms(proposals_single_left, scores_single.squeeze(1), nms_thresh)
RuntimeError: Not compiled with GPU support (nms at /home/yuhengq/Stereo-RCNN/lib/model/csrc/nms.h:22)

When python3 setup.py build develop
No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda' running install

hi,i've met the same problem with you, could you please tell me how did you fix it? thanks in advance

@Hexuanfang It has been a long time. I am sorry I can't remember all details clearly.
I think this is a Cuda configuration or installation issue. Try to verify your cuda version and installation. BTW, I suggest you configure this with conda, which definitely saves your time.