python - Dlib 未在 Google Colab 上使用 GPU

Question

如何强制在 GPU 上进行训练？

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

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

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

注意：GPU 设置为 RUN

标签： pythongoogle-colaboratorydlib

Answer 1

评论：

显然，正如我所测试的那样，这不是训练错误，而是加载错误。加载 ibug-300W 文件需要花费大量时间和内存。有什么办法可以更快地加载吗？

如果有人在 google colab 上偶然发现了这个问题或问题（训练时间慢）。

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

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