使用dlib和gpu进行分段错误

时间:2017-01-11 09:13:20

标签: c++ segmentation-fault gpu dlib

我刚刚安装了dlib,我使用cpu introduction example创建了一个项目,使用gpu introduction example创建了另一个项目。第一个是在我的计算机上编译和执行的,第二个是在Linux服务器上编译和执行的,以使用Nvidia GC。对于cpu示例,我可以毫无问题地编译和执行。对于gpu示例,我可以编译没有任何问题但是当我执行它时,初始化是正常的,但是当网络正在训练时,我已经正确完成了第一个时期,然后我遇到了分段错误。使用的方法是trainer.train_one_step(mini_batch_samples, mini_batch_labels);

我还尝试使用cpu训练网络,然后使用deserialize方法将其导入到gpu示例中,但在尝试进行预测时我也遇到了分段错误。用于预测的方法是std::vector<unsigned long> predicted_labels = tnet(training_images);

即使没有更改示例代码中的任何内容,我也遇到此问题,因此我不明白为什么会出现此段错误。如果有人可以帮助我,我将非常感激。

1 个答案:

答案 0 :(得分:0)

我为这个问题道歉但是为了防止这个真正愚蠢的错误发生在某人身上,我没有安装最新版本的dlib,这引起了麻烦。安装最新版本后,它正常工作......