Dlib未在Google Colab上使用GPU

问题描述

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

当前,即使我运行dlib.DLIB_USE_CUDA并且显示true,它也仅使用cpu

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

下面的图片显示了实际上我在运行代码时,GPU上没有任何运行:

enter image description here

注意:GPU设置为RUn

解决方法

评论:

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

如果有人在google colab上偶然发现了此问题或问题(培训时间缓慢)。

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

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