search

sanghoon / pva-faster-rcnn

Demo code for PVANet
https://arxiv.org/abs/1611.08588
Other
650 stars 241 forks source link

Initialization fails in GPU #101

Open shridharkini6 opened 6 years ago

shridharkini6 commented 6 years ago

Hi i am trying to run ./tools/test_pvanet.py. It works fine in CPU but in GPU it fails. figured out that im_detect function fails with F0213 23:08:03.355202 19911 roi_pooling_layer.cu:88] Check failed: error == cudaSuccess (48 vs. 0) no kernel image is available for execution on the device . GPU : GT 750M CUDA: 9.1 CUDNN: 7.0.5 OS:Ubuntu 16.04 Thanks