the problem of test

[hua1024]

when i run : python3 test_RFB.py -d VOC -v RFB_vgg -s 300 --trained_model ./weights/Final_RFB_vgg_VOC.pth , I have a problem , like File "/media/vision/HD_2TB/cvision/RFBNet/utils/nms_wrapper.py", line 9, in from .nms.gpu_nms import gpu_nms ImportError: libcudart.so.10.0: cannot open shared object file: No such file or directory

envinorment : pytorch1.0+cuda9.2+nvdia396 , i can find libcudart.so.9.2 in /usr/local/cuda/lib64

[lofyol]

me too，have you solved it ？

[hua1024]

I can not solve it , finally , use cpu nms

[zeroorhero]

hello, have you solved it?