Dlib 不在 Google Colab 上使用 GPU

Question

如何在 GPU 上强制训练？

目前即使我运行它也只使用 CPU dlib.DLIB_USE_CUDA，它说true。

它还说1当我跑步时print(dlib.cuda.get_num_devices())

这是附加的图像，显示实际上我正在运行代码时 GPU 上没有运行任何东西：

Answer 1

评论：显然，正如我所测试的那样，这不是训练错误，而是加载错误。加载 ibug-300W 文件需要很多时间和内存。有什么办法可以更快地加载它吗？如果有人在 google colab 上偶然发现了这个问题或问题（训练时间慢）。加载速度更快的方法是直接在 colab 的 vm/content 上传输数据集。因为 Drive 和 Colab 之间的传输速度很慢。PS：你需要至少 14-15GB 的内存来加载 ibug-300W 文件。