-
Notifications
You must be signed in to change notification settings - Fork 20
Description
I want to train it by:
bash experiments/scripts/train_faster_rcnn.sh 0,1,2,3 pascal_voc vgg16
but found an error:
Traceback (most recent call last):
File "./tools/trainval_net.py", line 21, in
from nets.vgg16 import MELM_vgg16
File "/home/share/CASD/tools/../lib/nets/vgg16.py", line 10, in
from nets.network import Network
File "/home/share/CASD/tools/../lib/nets/network.py", line 18, in
from ops.roi_pool import RoIPool
File "/home/share/CASD/tools/../lib/ops/init.py", line 13, in
from .roi_ring_pool import RoIRingPool
File "/home/share/CASD/tools/../lib/ops/roi_ring_pool/init.py", line 9, in
from .roi_ring_pool import RoIRingPool
File "/home/share/CASD/tools/../lib/ops/roi_ring_pool/roi_ring_pool.py", line 13, in
from . import roi_ring_pool_cuda
ImportError: libcudart.so.10.1: cannot open shared object file: No such file or directory
it makes my crazy,how to solve it (ㄒoㄒ)