pva-faster-rcnn
pva-faster-rcnn copied to clipboard
Initialization fails in GPU
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