the problem of test
hua1024 opened this issue · 3 comments
hua1024 commented
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 commented
me too,have you solved it ?
hua1024 commented
I can not solve it , finally , use cpu nms
zeroorhero commented
hello, have you solved it?