Dlib在Google Colab上没有使用GPU

Question

如何强制使用GPU进行训练？

即使我运行 dlib.DLIB_USE_CUDA 并且显示为 true，目前它仍然只使用CPU。

当我运行 print(dlib.cuda.get_num_devices()) 时，它也显示为 1。

下面是附加的图片，显示在代码实际运行时没有任何东西在GPU上运行：

注意：已将GPU设置为“运行”状态。

- xsephtion

---

显然，根据我的测试这不是训练错误，而是加载错误。 加载Ibug文件需要很长时间和大量内存。 有没有办法更快地加载它？ - xsephtion

Answer 1

评论：

根据我的测试，这似乎不是训练错误，而是加载错误。加载ibug-300W文件需要大量时间和内存。有没有办法加快加载速度？

如果有人在Google Colab上遇到此问题或问题（训练时间慢）。

加快加载速度的方法是直接将数据集传输到Colab的vm/content上。因为Drive和Colab之间的传输速度很慢。

附：您需要至少14-15GB的内存才能加载ibug-300W文件。

- xsephtion