Dlib 未在 Google Colab 上使用 GPU答案

【问题标题】：Dlib not using GPU on Google ColabDlib 未在 Google Colab 上使用 GPU
【发布时间】：2020-12-04 05:53:21
【问题描述】：

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

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

当我运行 print(dlib.cuda.get_num_devices()) 时，它还会显示 1

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

注意：GPU 设置为 RUN

【问题讨论】：

【解决方案1】：

评论：

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

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

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

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

【讨论】：