我在使用AWS SageMaker进行yolo训练的jupyter-notebook时遇到了困难。
我想要使用darknet模型开始训练,但是它并没有很好地工作。
我尝试了下面这些代码,所有代码都运行良好。
! conda install cudatoolkit -y
! conda install cudnn -y
! conda install -c fragcolor cuda10.0 -y
! conda update --all -y
我尝试训练模型...
! ./darknet detector train lamp.data yolov3-lamps.cfg darknet53.conv.74 -gpus 0,1,2,3,4,5,6,7
但是出现了这个错误。
./darknet: error while loading shared libraries: libcudart.so.10.0: cannot open shared object file: No such file or directory
我该如何解决这个问题?